2022年6月27日

pytorch中网络参数的默认精度

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

posted @ 2022-06-27 11:29 那抹阳光1994 阅读(3245) 评论(0) 推荐(0) 编辑

导航