Stefan Wolf commited on
Commit
17c2c1f
1 Parent(s): a141a99

Don't use parallel data loader to ease testing.

Browse files
models/swinv2_base_w24_b16x4-fp16_fungi+val_res_384_cb_epochs_6.py CHANGED
@@ -209,9 +209,9 @@ test_dataloader = dict(
209
  dict(type='PackInputs'),
210
  ],
211
  type='FungiTest'),
212
- num_workers=8,
213
- persistent_workers=True,
214
- pin_memory=True,
215
  sampler=dict(shuffle=False, type='DefaultSampler'))
216
  test_evaluator = dict(
217
  items=[
 
209
  dict(type='PackInputs'),
210
  ],
211
  type='FungiTest'),
212
+ num_workers=0,
213
+ persistent_workers=False,
214
+ pin_memory=False,
215
  sampler=dict(shuffle=False, type='DefaultSampler'))
216
  test_evaluator = dict(
217
  items=[