Solved by Nikronic in post #21 
              
              
                Hi, 
Try this in the line you defined Hq = self.lq(hq): 
self.lq = self.lq.float()
Hq = self.lq(hq)

Similar case: 
# no error
linear = nn.Linear(1, 5).half().cuda()
x = torch.randn(1, 5, 1).cuda()
linear = linear.float()
linear(x)


# your error
linear = nn.Linear(1, 5).half().cuda()
x = torch.rand…