写文章
- @zcshiguangji
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
使用Pytorch构建网络时，损失函数需要迁移到GPU上吗？
再进一步，如CrossEntropyCriterion等pytorch自带的loss类(或称losslayer)虽然也是Module的子类，但是这些类的成员变量不含Parameter对象或Buffer对象，所以无需to或cuda处理。模型输入tensor是数据不用多说，Module对象使用.cuda是为了将模型内部的数据(即每个子叶Module的Parameter或Buffer成员变量)递归地传入
到底了