From d304874dd7cbcc5bab9645dfd25711704bb360ec Mon Sep 17 00:00:00 2001 From: EmmaRenauld Date: Mon, 18 Mar 2024 10:44:02 -0400 Subject: [PATCH] Print in trainer's params --- dwi_ml/training/trainers.py | 1 + 1 file changed, 1 insertion(+) 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,