Skip to content

Commit

Permalink
fix: move validation dataloader check in PyTorchTrial [MD-515] (#9923)
Browse files Browse the repository at this point in the history
  • Loading branch information
azhou-determined authored Sep 12, 2024
1 parent db2881f commit fd7f77a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion harness/determined/pytorch/_pytorch_trial.py
Original file line number Diff line number Diff line change
Expand Up @@ -992,8 +992,8 @@ def _validate(self, searcher_op: Optional[core.SearcherOperation] = None) -> Dic

else:
assert self._evaluate_full_dataset_defined(), "evaluate_full_dataset not defined."
assert self.validation_loader is not None
if self.is_chief:
assert self.validation_loader is not None
metrics = self.trial.evaluate_full_dataset(data_loader=self.validation_loader)

if not isinstance(metrics, dict):
Expand Down

0 comments on commit fd7f77a

Please sign in to comment.