Shuo Zhao

When I use Multi-GPU for Training, it has error:in dump ForkingPickler(file, protocol).dump(obj) AttributeError: Can't pickle local object 'TorchHistory.add_log_parameters_hook..'

https://github.com/HealthML/active-segmentation/blob/main/src/datasets/doubly_shuffled_nifti_dataset.py#L499 this line will throw an error when self.dim is not 2.