Change num of epochs

main.py

@@ -12,7 +12,7 @@ if __name__ == '__main__':
     # train
     parser.add_argument('--lr', type=float, default=2e-4)
     parser.add_argument('--weight_decay', type=float, default=2.5e-5)
-    parser.add_argument('--epochs', type=int, default=512)
+    parser.add_argument('--epochs', type=int, default=100)
     parser.add_argument('--batch_size', type=int, default=128)
     parser.add_argument('--betas', type=float, nargs='+', default=(.5, .999))
     # misc