I have tried using this method here:

weight_tensor = model_gpu.conv1.weight.data.cpu()
a = weight_tensor[0] #
print(a)
print(a.size())
plt.imshow(a[0, …])

But I keep getting the error:

(0.0, 351.0, ‘0’, ‘black’, ‘center_baseline’, ‘right’, -7373694105942136849, None, None, 72.0, <weakref at 0x7f6aeb629f98; dead>, 1.2)