Stage 1 now works fine. But in stage 2 I have this error: "Input type (torch.cuda.FloatTensor) and w
Stage 1 now works fine. But in stage 2 I have this error: "Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same"

