diff --git a/src/hyperactive/experiment/integrations/torch_lightning_experiment.py b/src/hyperactive/experiment/integrations/torch_lightning_experiment.py
index 0bdd1f80..ae4a8f86 100644
--- a/src/hyperactive/experiment/integrations/torch_lightning_experiment.py
+++ b/src/hyperactive/experiment/integrations/torch_lightning_experiment.py
@@ -177,7 +177,10 @@ def _evaluate(self, params):
             trainer.fit(model, self.datamodule)
 
             val_result = trainer.callback_metrics.get(self.objective_metric)
-            metadata = {}
+            metadata = {= {
+            "num_epochs_trained": trainer.current_epoch,
+            "all_metrics": trainer.callback_metrics,
+        }
 
             if val_result is None:
                 available_metrics = list(trainer.callback_metrics.keys())