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

2019-04-16 15:23:21 -07:00
parent 3d78e226e6
commit 87677fcc4d
1 changed files with 1 additions and 1 deletions
--- a/examples/run_gpt2.py
+++ b/examples/run_gpt2.py
@@ -58,7 +58,7 @@ def run_model():
    parser.add_argument("--nsamples", type=int, default=1)
    parser.add_argument("--batch_size", 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('--unconditional', action='store_true', help='If true, unconditional generation.')
    args = parser.parse_args()