We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9f67ad7 commit d8a1fe5Copy full SHA for d8a1fe5
config.py
@@ -35,7 +35,7 @@ def add_argument_group(name):
35
train_arg = add_argument_group('Training')
36
train_arg.add_argument('--is_train', type=str2bool, default=True)
37
train_arg.add_argument('--optimizer', type=str, default='adam')
38
-train_arg.add_argument('--max_step', type=int, default=50000)
+train_arg.add_argument('--max_step', type=int, default=20000)
39
train_arg.add_argument('--lr', type=float, default=0.0001)
40
train_arg.add_argument('--beta1', type=float, default=0.5)
41
train_arg.add_argument('--beta2', type=float, default=0.999)
0 commit comments