change dropout input type

ahmedbesbes · Oct 22, 2019 · a8559da · a8559da
1 parent 88f7c30
commit a8559da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -367,7 +367,7 @@ def run(args, both_cases=False):
     parser.add_argument('--number_of_characters', type=int, default=69)
     parser.add_argument('--extra_characters', type=str, default='')
     parser.add_argument('--max_length', type=int, default=150)
-    parser.add_argument('--dropout_input', type=int, default=0.1)
+    parser.add_argument('--dropout_input', type=float, default=0.1)
     parser.add_argument('--epochs', type=int, default=10)
     parser.add_argument('--batch_size', type=int, default=128)
     parser.add_argument('--optimizer', type=str,