Fly2leo

赞

0

|

搜索文章

3年前

PyTorch 多 GPU 训练的模型权重 key中多 ‘module.‘

使用 PyTorch 的 distributed 工具执行多 GPU 训练保存模型后，权重的 key 多了 'module.'，再次加载 state_dict 会出现 ‘“...

0

评论

个人成就

文章被阅读 1,001

加入于

2022-08-29