I am a pytorch beginer,and recnently when I try to translate a network model defined by myself,
I met a problem.
the problem can be described simply like this :
first I define a class
class cliqueBlock(nn.moudle): def __init__(self,out_channels = 36): super(cliqueBlock,self).__init() self.w0 = w0 # here w0 is a list of nn.moudle like [ nn.ReLU(), nn.Conv2d(),nn.BatchNorm2d()] def forward(self,x) out = self.w0(x) return out
then I transfer the model to cuda : model = cliqueBlock.cuda()
the problem is , when I use this model to train my network ,
RuntimeError: get_device is not implemented for type torch.FloatTensor
I think this is due to my image input is in cuda while the paramters of my networks is in cpu
but I just confused about the reason, should not I define the class member in init as a list ?
and any solution to this problem ?
thank you very much