Added files

modules/trainer.py

@@ -103,6 +103,7 @@ def train_and_test_model(
 
     # Save the model using torch save as backup
     print("Saving the model.")
+    print(f"Model saved to {config.MODEL_PATH}")
     create_folder_if_not_exists(config.MODEL_PATH)
     torch.save(model.state_dict(), config.MODEL_PATH)
 

modules/utils.py

@@ -63,6 +63,7 @@ def create_folder_if_not_exists(file_path):
     """
     # Extract the folder path
     folder_path = os.path.dirname(file_path)
+    print(f"Folder path: {folder_path}")
 
     # Create the folder if it does not exist
     if not os.path.exists(folder_path):