diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-09 23:24:02 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-09 23:24:02 +0200 |
| commit | 53677be4ec14854ea4881b0d78730e0414c8dedd (patch) | |
| tree | 56eaace5e9906c7d408b6a251ca100b5c8b4e991 /src/training/train.py | |
| parent | 125d5da5fb845d03bda91426e172bca7f537584a (diff) | |
Working bash scripts etc.
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() |