HealthML / active-segmentation

ActiveSegmentation: A Simulation Framework for Benchmarking Active Learning Strategies for 3D Medical Image Segmentation
GNU Affero General Public License v3.0
16 stars 1 forks source link

Multi-GPU Training Issues #160

Open funkyZhaofour opened 1 year ago

funkyZhaofour commented 1 year ago

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..'