golemz

golemz

威望 : 0 赞同 : 3 感谢 : 0

更多 »回复

0

假设有4个标签,模型里面__init__最后两步:         self.out = nn.ModuleList()         for di in range(4):             self.out.append(nn.Linear(HID...

0

向量维度弄错了,没有考虑batch_size吧,最后的self.out的输入应该是256 x 256,而不是32768 x 2

0

这是windows下0.4.0之前的版本的问题吧,num_workers=2改成0试试.

0

你这是一次把所有图片读进显存再保存,显存消耗当然大,另外,既然要用np保存,为什么要调用cuda放显存里?

更多 »发问

5

1843 次浏览  • 3 个关注   • 2018-01-11

发问

回复

文章

最新动态

详细资料

个人成就:

威望: 0 赞同: 3 感谢: 0

最后活跃:
1 秒前
更多 » 关注 1

ptorch

更多 » 0 人关注
关注 2 话题
主页访问量 : 714 次访问