fix: NoneType epochs

2025-12-10 15:37:54 +01:00 · 2025-12-10 15:37:54 +01:00 · 6c5908e6ae
commit 6c5908e6ae
parent 63119980c9
4 changed files with 6 additions and 4 deletions
--- a/src/train.py
+++ b/src/train.py
@ -1,7 +1,6 @@
 from pathlib import Path

 import torch
-from torch import nn
 from torch.utils.data import DataLoader

 from src.dataset_loaders import dataset_called