This is what I have-
images = Variable(torch.from_numpy(X)).to(device) # [batch, channel, H, W]
masks = Variable(torch.from_numpy(y)).to(device)
# print("images",type(images)) # output: tensor class
images = Normalize(images,(0, 0, 0, 0, 0), (0, 0, 0,0, 0))
masks = Normalize(masks, (0.0), (0.0))
optim.zero_grad()
outputs = model(images)
loss = criterion(outputs, masks)
loss.backward()
I get the error TypeError: conv2d(): argument 'input' (position 1) must be Tensor, not Normalize
To the best of my understanding, model
needs a tensor. How should I give it that?