06 2022 档案

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

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

导航

点击右上角即可分享
微信分享提示