
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
x.to(‘cuda‘)或者x.device = ‘cuda‘没有用,无法将数据从CPU放到GPU
我通过尝试用x = x.to('cuda')解决。具体为什么我也不清楚,试出来的。直接使用x.to('cuda')没用,这时候代码理论上是没问题的。

pytorch保存和加载模型权重的方式
这种方式需要自己定义网络,并且其中的参数名称与结构要与保存的模型中的一致(可以是部分网络,比如只使用VGG的前几层),相对灵活,便于对网络进行修改。这种方式无需自定义网络,保存时已把网络结构保存,比较死板,不能调整网络结构。仅作为个人学习笔记使用。

VSRN论文阅读笔记:Visual Semantic Reasoning for Image-Text Matching
VSRN论文阅读笔记:Visual Semantic Reasoning for Image-Text Matching

到底了







