changes to training + added autoencoder

src/train.py

@@ -26,7 +26,7 @@ def train(
     assert model_name or model_path, "Either a model to train or a model to load from model_path must be provided"
 
     if model_name:
-        print("Creating model")
+        print(f"Creating model: {model_name}")
         model = model_called[model_name]
     else:
         print("Loading model from disk")
@@ -64,6 +64,7 @@ def train(
     print("Training")
     best_model = trainer.execute(
         model=model,
+        context_length=context_length,
         train_loader=training_loader,
         validation_loader=validation_loader,
         n_epochs=n_trials,