initial commit

mkocabas · Oct 12, 2021 · c7b25d9 · c7b25d9
1 parent bc7dc38
commit c7b25d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts/spec_eval.py b/scripts/spec_eval.py
@@ -26,6 +26,7 @@ def main(hparams):
 
  logger.info(f'Hyperparameters: \n {hparams}')
 
+ hparams.DATASET.NUM_WORKERS = -1 # 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