2019年7月26日
pytorch的state_dict()拷贝问题
摘要： 先说结论，model.state_dict()是浅拷贝，返回的参数仍然会随着网络的训练而变化。应该使用deepcopy(model.state_dict())，或将参数及时序列化到硬盘。 再讲故事，前几天在做一个模型的交叉验证训练时，通过model.state_dict()保存了每一组交叉验证模型的 阅读全文
posted @ 2019-07-26 09:33 Luke_Ye 阅读(4291) 评论(0) 推荐(1) 编辑