trying 2070 with 512

KasuganoLove · Mar 31, 2022 · 75d0896 · 75d0896
1 parent 4213822
commit 75d0896
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -30,13 +30,13 @@
 # parser.add_argument('-crop_size', help='Set the crop_size', default=[256, 256], nargs='+', type=int)
 parser.add_argument('-crop_size', help='Set the crop_size', default=[512, 512], nargs='+', type=int)
 # parser.add_argument('-train_batch_size', help='Set the training batch size', default=16, type=int)
-parser.add_argument('-train_batch_size', help='Set the training batch size', default=8, type=int)
+parser.add_argument('-train_batch_size', help='Set the training batch size', default=10, type=int)
 parser.add_argument('-epoch_start', help='Starting epoch number of the training', default=0, type=int)
 parser.add_argument('-lambda_loss', help='Set the lambda in loss function', default=0.04, type=float)
 parser.add_argument('-val_batch_size', help='Set the validation/test batch size', default=1, type=int)
 parser.add_argument('-exp_name', help='directory for saving the networks of the experiment', default="ckpt", type=str)
 parser.add_argument('-seed', help='set random seed', default=19, type=int)
-parser.add_argument('-num_epochs', help='number of epochs', default=200, type=int)
+parser.add_argument('-num_epochs', help='number of epochs', default=1000, type=int)
 
 args = parser.parse_args()