Skip to content

Commit d8a1fe5

Browse files
author
kmbae
committed
Reduce max iter
1 parent 9f67ad7 commit d8a1fe5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def add_argument_group(name):
3535
train_arg = add_argument_group('Training')
3636
train_arg.add_argument('--is_train', type=str2bool, default=True)
3737
train_arg.add_argument('--optimizer', type=str, default='adam')
38-
train_arg.add_argument('--max_step', type=int, default=50000)
38+
train_arg.add_argument('--max_step', type=int, default=20000)
3939
train_arg.add_argument('--lr', type=float, default=0.0001)
4040
train_arg.add_argument('--beta1', type=float, default=0.5)
4141
train_arg.add_argument('--beta2', type=float, default=0.999)

0 commit comments

Comments
 (0)