num_workers=7

new_test_saved_finetuned_model.py

@@ -555,7 +555,7 @@ def train():
     parser.add_argument("-b", "--batch_size", type=int, default=500, help="number of batch_size") #64
     parser.add_argument("-e", "--epochs", type=int, default=1)#1501, help="number of epochs") #501
     # Use 50 for pretrain, and 10 for fine tune
-    parser.add_argument("-w", "--num_workers", type=int, default=0, help="dataloader worker size")
+    parser.add_argument("-w", "--num_workers", type=int, default=7, help="dataloader worker size")
 
     # Later run with cuda
     parser.add_argument("--with_cuda", type=bool, default=False, help="training with CUDA: true, or false")