把在forward函数里面添加的层放到GPU中

问题描述

借鉴GoogleNet和NiN的思想，我想在Inception最后在添加两个卷积全连接层，但是这样子写会报错，说是final这层的权重没有在GPU中。

问题解决

分为两步：
(1) 手动把self.final层的参数加入到GPU中

(2)把net先手动加入到GPU中，并且先在GPU中跑一下，因为只有在GPU中运行了一下，这个self.final才会到GPU中。

一开始实例化一个模型是不会把forward函数里面定义的层加进去的

只有跑了才会加进入

所以把X加入到GPU中，然后把模型加入到GPU中先预热一下，自己定义的层就会出现在模型中了。