问题描述
借鉴GoogleNet和NiN的思想,我想在Inception最后在添加两个卷积全连接层,但是这样子写会报错,说是final这层的权重没有在GPU中。
问题解决
分为两步:
(1) 手动把self.final层的参数加入到GPU中
(2)把net先手动加入到GPU中,并且先在GPU中跑一下,因为只有在GPU中运行了一下,这个self.final才会到GPU中。
一开始实例化一个模型是不会把forward函数里面定义的层加进去的
只有跑了才会加进入
所以把X加入到GPU中,然后把模型加入到GPU中先预热一下,自己定义的层就会出现在模型中了。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)