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:
param.optimizer.step()
return loss / len(data_loader)
def _eval_epoch(self, phase: str):
def _eval_epoch(self):
torch.set_grad_enabled(False)
for model in self.models:
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