Skip to content
Snippets Groups Projects
Commit 2da7d4f2 authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

eval_epoch no longer requires the phase argument

parent 2f7abaaa
No related branches found
No related tags found
No related merge requests found
...@@ -110,7 +110,7 @@ class AbstractTraining: ...@@ -110,7 +110,7 @@ class AbstractTraining:
param.optimizer.step() param.optimizer.step()
return loss / len(data_loader) return loss / len(data_loader)
def _eval_epoch(self, phase: str): def _eval_epoch(self):
torch.set_grad_enabled(False) torch.set_grad_enabled(False)
for model in self.models: for model in self.models:
model.eval() model.eval()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment