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

Fix loss logging #17

Closed desi-ivanov closed 1 year ago

desi-ivanov commented 1 year ago

The epoch loss is wrongly calculated as it includes only the last batch loss