from torchsummary import summary
summary(net, input_size=(1, input_size))
RuntimeError Traceback (most recent call last) in () 1 2 from torchsummary import summary ----> 3 summary(net, input_size=(1, input_size))
5 frames /usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in linear(input, weight, bias) 1610 ret = torch.addmm(bias, input, weight.t()) 1611 else: -> 1612 output = input.matmul(weight.t()) 1613 if bias is not None: 1614 output += bias
RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat2' in call to thmm
