Hey guys! Could you all help me with this issue?
“RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same”
I am using this function to initialize the weights:
def weights_init(model):
    if type(model) in [nn.Conv2d, nn.Linear]:
        nn.init.xavier_normal_(model.weight.data)
        nn.init.constant_(model.bias.data, 0.1)

And I am using this code to submit an image to the CNN layers:
for inputs, labels in images_loader:    
    inputs = inputs.to(device) 
    labels = labels.to(device) 
    inputs = inputs.unsqueeze(0)    

    print(inputs.shape)
    print(labels.shape)
    
    outputs = model(inputs)

Best regards,
Matheus Santos.