摘要:
可以调用模型的父类Module中的type方法,例如model.type(torch.float64),将网络模型model的参数和缓冲区强制转换为torch.float64类型,这样就可以训练torch.float64类型的数据了,还可以指定其他类型。另外还有一些强制转换为某一种类型的方法:flo 阅读全文
posted @ 2023-12-06 15:50
ZephyrYin
阅读(322)
评论(0)
推荐(0)
浙公网安备 33010602011771号