pytorch学习
zcshiguangji
这个作者很懒,什么都没留下…
展开
-
使用Pytorch构建网络时,损失函数需要迁移到GPU上吗?
再进一步,如CrossEntropyCriterion等pytorch自带的loss类(或称losslayer)虽然也是Module的子类,但是这些类的成员变量不含Parameter对象或Buffer对象,所以无需to或cuda处理。模型输入tensor是数据不用多说,Module对象使用.cuda是为了将模型内部的数据(即每个子叶Module的Parameter或Buffer成员变量)递归地传入gpu。...转载 2022-08-01 10:22:20 · 1218 阅读 · 1 评论 -
图像读取格式转换
图像读取格式tensor numpy PIL相互转换原创 2022-07-26 11:03:21 · 217 阅读 · 1 评论