pytorch中网络参数的默认精度
摘要:pytorch默认使用单精度float32训练模型,其主要原因为:使用float16训练模型,模型效果会有损失,而使用double(float64)会有2倍的内存压力,且不会带来太多的精度提升,因此默认使用单精度float32训练模型。 由于输入类型不一致导致报错: PyTorch:expected
阅读全文
posted @ 2022-06-27 11:29
posted @ 2022-06-27 11:29