diff options
Diffstat (limited to 'src/training/train.py')
| -rw-r--r-- | src/training/train.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/training/train.py b/src/training/train.py index 3334c2e..aaa0430 100644 --- a/src/training/train.py +++ b/src/training/train.py @@ -106,7 +106,7 @@ class Trainer: # Running average for the loss. loss_avg = RunningAverage() - data_loader = self.model.data_loaders("train") + data_loader = self.model.data_loaders["train"] with tqdm( total=len(data_loader), @@ -164,7 +164,7 @@ class Trainer: self.model.eval() # Running average for the loss. - data_loader = self.model.data_loaders("val") + data_loader = self.model.data_loaders["val"] # Running average for the loss. loss_avg = RunningAverage() |