Skip to content

S
SCL-my
Commit d2169327 by yuxguo
Options

fix

parent 3af37a61
Showing with 2 additions and 2 deletions
main.py
...@@ -102,9 +102,9 @@ trainer_args.add_argument('--epochs', '-e', type=int, default=200, ...@@ -102,9 +102,9 @@ trainer_args.add_argument('--epochs', '-e', type=int, default=200,
help='the number of epochs') help='the number of epochs')
trainer_args.add_argument('--obs-epochs', '-oe', type=int, default=5, trainer_args.add_argument('--obs-epochs', '-oe', type=int, default=5,
help='the number of sub epochs for observation stage') help='the number of sub epochs for observation stage')
trainer_args.add_argument('--batch-size', '-bs', type=int, default=32, trainer_args.add_argument('--batch-size', '-bs', type=int, default=8,
help='input batch size for training (default: 128)') help='input batch size for training (default: 128)')
trainer_args.add_argument('--eval-batch-size', '-ebs', type=int, default=32, trainer_args.add_argument('--eval-batch-size', '-ebs', type=int, default=8,
help='input batch size for evaluation (default: 32)') help='input batch size for evaluation (default: 32)')
trainer_args.add_argument('--lr', '-lr', type=float, default=0.005, trainer_args.add_argument('--lr', '-lr', type=float, default=0.005,
help='learning rate (default: 0.005)') help='learning rate (default: 0.005)')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment