search

lucidrains / med-seg-diff-pytorch

Implementation of MedSegDiff in Pytorch - SOTA medical segmentation using DDPM and filtering of features in fourier space
MIT License
199 stars 29 forks source link

multi gpu #12

Closed TMullerSG closed 1 year ago

TMullerSG commented 1 year ago

when use accelator for multiple gpu, i need to make the follow changes in med_seg_diff_pytorch.py

self.model = model.module
self.input_img_channels = self.model.input_img_channels
self.mask_channels = self.model.mask_channels
self.self_condition = self.model.self_condition
self.image_size = self.model.image_size
lucidrains commented 1 year ago

@TMullerSG oh thanks! sorry i missed this

made the changes you recommended