changes to training + added autoencoder

2025-12-13 17:53:01 +01:00 · 2025-12-13 17:53:01 +01:00 · a4a41d190b
commit a4a41d190b
parent 6e591bb470
7 changed files with 91 additions and 55 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@ -10,7 +10,8 @@ class Trainer(ABC):
    @abstractmethod
    def execute(
            self,
-            model: nn.Module | None,
+            model: nn.Module | type[nn.Module] | None,
+            context_length: int,
            train_loader: DataLoader,
            validation_loader: DataLoader,
            n_epochs: int | None,