dsgt-kaggle-clef
/

dsgt-snakeclef

Model card Files Files and versions Community

Anthony Miyaguchi commited on May 24

Commit

2759b86

•

1 Parent(s): d83944d

Remove num_workers

Files changed (1) hide show

script.py +1 -1

script.py CHANGED Viewed

@@ -63,7 +63,7 @@ def make_submission(
     model.load_state_dict(checkpoint["state_dict"])
     dataloader = DataLoader(
-        ImageDataset(test_metadata, images_root_path), batch_size=32, num_workers=1
     )
     rows = []
     for batch in dataloader:

     model.load_state_dict(checkpoint["state_dict"])
     dataloader = DataLoader(
+        ImageDataset(test_metadata, images_root_path), batch_size=32
     )
     rows = []
     for batch in dataloader: