diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py index fb3aa5d28..4b7813b1d 100644 --- a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py +++ b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py @@ -279,12 +279,18 @@ def _save_debug_information(self): # print(k) pass if k in ['dataloader_train', 'dataloader_val']: - if hasattr(getattr(self, k), 'generator'): - dct[k + '.generator'] = str(getattr(self, k).generator) - if hasattr(getattr(self, k), 'num_processes'): - dct[k + '.num_processes'] = str(getattr(self, k).num_processes) - if hasattr(getattr(self, k), 'transform'): - dct[k + '.transform'] = str(getattr(self, k).transform) + dl = getattr(self, k) + if hasattr(dl, 'generator'): + dct[k + '.generator'] = str(dl.generator) + if hasattr(dl.generator, 'transforms'): + try: + dct[k + '.generator.transforms'] = str(dl.generator.transforms) + except Exception as e: + dct[k + '.generator.transforms'] = f"Could not stringify generator.transforms: {type(e).__name__}: {e}" + if hasattr(dl, 'num_processes'): + dct[k + '.num_processes'] = str(dl.num_processes) + if hasattr(dl, 'transform'): + dct[k + '.transform'] = str(dl.transform) import subprocess hostname = subprocess.getoutput(['hostname']) dct['hostname'] = hostname