fix

main.py

@@ -102,9 +102,9 @@ trainer_args.add_argument('--epochs', '-e', type=int, default=200,
     help='the number of epochs')
 trainer_args.add_argument('--obs-epochs', '-oe', type=int, default=5,
     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)')
-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)')
 trainer_args.add_argument('--lr', '-lr', type=float, default=0.005,
     help='learning rate (default: 0.005)')