about grad_scaler is set None

MrBlankness / LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"

https://arxiv.org/abs/2403.05246

Apache License 2.0

279 stars 25 forks source link

about grad_scaler is set None #28

Open Xchhher opened 3 months ago

Xchhher commented 3 months ago

hi, thx for the great work! I have a question about the code. In the class "nnUNetTrainerLightMUNet", I noticed that you set "self.grad_scaler = None" In the case of using CUDA, Is there some special reason?