SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[run_gpt2.py] temperature should be a float, not int

Browse Source
This commit is contained in:
Ben Mann
2019-04-16 15:23:21 -07:00
committed by GitHub
parent 3d78e226e6
commit 87677fcc4d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/run_gpt2.py
View File

@@ -58,7 +58,7 @@ def run_model():
parser.add_argument("--nsamples", type=int, default=1) parser.add_argument("--nsamples", type=int, default=1)
parser.add_argument("--batch_size", type=int, default=-1) parser.add_argument("--batch_size", type=int, default=-1)
parser.add_argument("--length", type=int, default=-1) parser.add_argument("--length", type=int, default=-1)
parser.add_argument("--temperature", type=int, default=1) parser.add_argument("--temperature", type=float, default=1.0)
parser.add_argument("--top_k", type=int, default=0) parser.add_argument("--top_k", type=int, default=0)
parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.') parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.')
args = parser.parse_args() args = parser.parse_args()