pytorch
pytorch相关记录
crasyter
这个作者很懒,什么都没留下…
展开
-
pytorch相关函数
这里对于dim维度求和,注意的是每一个元素的值都是与之前元素的求和。4、tensor转化为numpy。3、创建全零矩阵tensor。原创 2022-11-25 20:49:43 · 558 阅读 · 0 评论 -
loss训练记录
参数丢失了 requires_grad属性,所以导致没能梯度更新。自定义的loss需要转成tensor形式。原创 2022-11-16 15:28:10 · 1697 阅读 · 0 评论 -
报错Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
其中 Input type 是我们喂进去的数据, weight type 是网络模型,看出前者位于CPU,后者由于model.cuda()已经在GPU上。原创 2022-11-14 16:44:00 · 1539 阅读 · 0 评论