search

constantinpape / torch-em

Deep-learning based semantic and instance segmentation for 3D Electron Microscopy and other bioimage analysis problems based on pytorch.
MIT License
69 stars 19 forks source link

Request for DistributedDataParallel #229

Open sladenheim opened 4 months ago

sladenheim commented 4 months ago

To allow for the training of larger models split across GPUS using the DefaultTrainer class, implement model parallel capabilities using the torch.DistributedDataParallel functionality.

constantinpape commented 4 months ago

Thanks for opening this @sladenheim. We are currently focusing on a new release of micro_sam, but will then look into this.