这是一个50层的卷积,加一个全连接层,但是我在gpu跑,他运算速度大概在5.7s,在cpu运算是2.7s,不应该是gpu比cpu快么。50层的卷积应该不是一个小的网络吧还有就是将网络的参数放在gpu的话,是不是都要在init 方法初始化参数?
1.还要看你的pytroch是不是GPU版本的?2.你的torch.tensor有没有放在GPU上