diff --git a/dwi_ml/training/trainers.py b/dwi_ml/training/trainers.py
index ed28fb47..57d56395 100644
--- a/dwi_ml/training/trainers.py
+++ b/dwi_ml/training/trainers.py
@@ -368,6 +368,7 @@ def save_params_to_json(self):
         with open(json_filename, 'w') as json_file:
             json_file.write(json.dumps(
                 {'Date': str(datetime.now()),
+                 'hdf5 file': self.batch_loader.dataset.hdf5_file,
                  'Trainer params': self.params_for_checkpoint,
                  'Sampler params': self.batch_sampler.params_for_checkpoint,
                  'Loader params': self.batch_loader.params_for_checkpoint,