Skip to content

Commit

Permalink
fixed an issue with eval forcing to have a val_dataloader in config, …
Browse files Browse the repository at this point in the history
…while for train it wasnt required (#1823)

Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
  • Loading branch information
NatanBagrov and BloodAxe committed Feb 9, 2024
1 parent 94cc2d6 commit 118d1cb
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/super_gradients/training/sg_trainer/sg_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,7 +353,9 @@ def evaluate_from_recipe(cls, cfg: DictConfig) -> Tuple[nn.Module, Tuple]:

# INSTANTIATE DATA LOADERS
val_dataloader = dataloaders.get(
name=cfg.val_dataloader, dataset_params=cfg.dataset_params.val_dataset_params, dataloader_params=cfg.dataset_params.val_dataloader_params
name=get_param(cfg, "val_dataloader"),
dataset_params=cfg.dataset_params.val_dataset_params,
dataloader_params=cfg.dataset_params.val_dataloader_params,
)

if cfg.checkpoint_params.checkpoint_path is None:
Expand Down

0 comments on commit 118d1cb

Please sign in to comment.