fix num_workers

mkocabas · Oct 14, 2021 · d2fe2c2 · d2fe2c2
1 parent ca3e7b0
commit d2fe2c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/spec_eval.py b/scripts/spec_eval.py
@@ -42,7 +42,7 @@ def main(hparams):
 
  logger.info(f'Hyperparameters: \n {hparams}')
 
- hparams.DATASET.NUM_WORKERS = -1 # set this to be compatible with other machines
+ hparams.DATASET.NUM_WORKERS = 0 # set this to be compatible with other machines
  model = SPECTrainer(hparams=hparams).to(device)
 
  # TRAINING.PRETRAINED_LIT points to the checkpoint files trained using this repo