Slight error in accelerator

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

Closed isamu-isozaki closed 1 year ago

isamu-isozaki commented 1 year ago

Sorry, found a slight bug where I did

loss.backward()

instead of

accelerator.backward(loss)

lucidrains commented 1 year ago

@isamu-isozaki thank you Isamu!

isamu-isozaki commented 1 year ago

Np!