Jul, 2023

混合精度加速 Fourier 神经算子

TL;DR通过对 Fourier neural operator(FNO)进行全精度和混合精度训练的内存和运行时时间进行分析,研究混合精度训练的数值稳定性,并设计了一种训练程序,有效减少了训练时间和内存使用,而在准确性上几乎没有减少,适用于 Navier-Stokes 和 Darcy 流动方程。