num_workers

src/test_saved_model.py

@@ -225,7 +225,7 @@ if __name__ == "__main__":
     print("Loading Test Dataset", args.test_dataset)
     test_dataset = TokenizerDataset(args.test_dataset, args.test_label, vocab_obj, seq_len=args.seq_len, train=False)
     print("Creating Dataloader")
-    test_data_loader = DataLoader(test_dataset, batch_size=args.batch_size, num_workers=10)
+    test_data_loader = DataLoader(test_dataset, batch_size=args.batch_size, num_workers=0)
     bert = torch.load(args.finetuned_bert_checkpoint, map_location="cpu")
         
     if args.workspace_name == "ratio_proportion_change4":