Skip to content

FNO: run spectral convolutions in float32 under autocast - #2150

Open
GeorgeLiang3 wants to merge 2 commits into
NVIDIA:mainfrom
GeorgeLiang3:fno-amp-fp32
Open

FNO: run spectral convolutions in float32 under autocast#2150
GeorgeLiang3 wants to merge 2 commits into
NVIDIA:mainfrom
GeorgeLiang3:fno-amp-fp32

test(fno): check gradients and weight updates in the autocast trainin…

92f5082
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs