pytorch
文章平均质量分 55
C1utch
杭州电专本科,未来就读西电杭研院硕士,研究方向计算成像。
展开
-
用Pytorch进行多GPU并行训练模型
主要着重讲dataparallel的使用方法。原创 2023-08-03 13:32:15 · 667 阅读 · 0 评论 -
(debug记录)pytorch导入模型时的error :size mismatch for fution.weight
完整报错:error :size mismatch for fution.weight: copying a param with shape torch.Size([28, 56, 1, 1]) from checkpoint, the shape in current model is torch.Size([28, 28, 1, 1])这是由于在加载模型的预训练权重时,遇到了一个size不匹配的问题,预训练模型和当前模型之间存在结构差异。这个子函数也可以去掉!原创 2023-04-11 15:48:16 · 1511 阅读 · 0 评论 -
TensorBorad+Pytorch一些使用记录
作为AI炼丹师的一些初级知识记录。原创 2023-04-19 17:39:21 · 220 阅读 · 0 评论