Update model and CI/CD workflow to support Python 3.9, 3.10, 3.11, an…

…d 3.12
VishwamAI · Jul 23, 2024 · ae67e38 · ae67e38
1 parent b36423b
commit ae67e38
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_train.py b/tests/test_train.py
@@ -89,6 +89,10 @@ def test_training_loop(self):
             print(f"History type: {type(history)}")
             print(f"History content: {history}")
             print(f"Actual number of epochs in history: {len(history)}")
+
+            print("Training history:")
+            for i, epoch_data in enumerate(history):
+                print(f"Epoch {i+1}: Train Loss: {epoch_data['train_loss']:.6f}, Val Loss: {epoch_data['val_loss']:.6f}")
         except Exception as e:
             self.fail(f"Training failed with error: {str(e)}")