Unscale gradients before clipping

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

GNU General Public License v3.0

9.32k stars 2.52k forks source link

Closed yelboudouri closed 9 months ago

yelboudouri commented 11 months ago

Gradients should be unscaled before clipping them. PyTorch Guide to use AMP