diff --git a/examples/poyo/configs/base.yaml b/examples/poyo/configs/base.yaml index 4fab6d2..fc3abb8 100644 --- a/examples/poyo/configs/base.yaml +++ b/examples/poyo/configs/base.yaml @@ -43,5 +43,4 @@ gpus: 1 ckpt_path: null # for resuming training, or loading a model during finetuning -fast_dev_run: false -num_sanity_val_steps: 0 \ No newline at end of file +num_sanity_val_steps: 0 # only supports 0 (none) and -1 (all) right now \ No newline at end of file diff --git a/examples/poyo/train.py b/examples/poyo/train.py index 9756064..663b729 100644 --- a/examples/poyo/train.py +++ b/examples/poyo/train.py @@ -43,7 +43,7 @@ def main(cfg: DictConfig): # setup loggers wandb_logger = None - if cfg.wandb.enable and not cfg.fast_dev_run: + if cfg.wandb.enable: wandb_logger = L.pytorch.loggers.WandbLogger( save_dir=cfg.log_dir, entity=cfg.wandb.entity, @@ -117,7 +117,6 @@ def main(cfg: DictConfig): devices=cfg.gpus, num_nodes=cfg.nodes, limit_val_batches=None, # Ensure no limit on validation batches - fast_dev_run=cfg.fast_dev_run, num_sanity_val_steps=cfg.num_sanity_val_steps, ) @@ -325,7 +324,7 @@ def val_dataloader(self): shuffle=False, batch_size=batch_size, collate_fn=collate, - num_workers=0, + num_workers=self.cfg.num_workers, drop_last=False, ) @@ -351,7 +350,7 @@ def test_dataloader(self): shuffle=False, batch_size=batch_size, collate_fn=collate, - num_workers=0, + num_workers=self.cfg.num_workers, ) self.log.info(f"Testing on {len(test_sampler)} samples")