Jan, 2024
通过混合精度加快科学机器学习速度和减少内存使用
Speeding up and reducing memory usage for scientific machine learning via mixed precision
Joel Hayford, Jacob Goldman-Wetzler, Eric Wang, Lu Lu
TL;DR科学机器学习 (SicML) 中的物理信息神经网络 (PINNs) 和深度算子网络 (DeepONets) 是用于解决偏微分方程的主要技术,但是使用混合精度训练这些网络既可以降低计算时间和内存消耗,又可以保持模型的准确性。