Hi,
I am trying to run the model in half-precision only. Sorry for the confusion.
My model seems to run for a few batches or even at times for an epoch but then it throughs error. The code of my model looks similar to the one below:
        images  = Variable(images).cuda()
        labels  = Variable(labels).cuda()


        net.half()  ## 
        images = images.half() ##

        #forward
        logits = net(images)
        logits = logits.float() ##

        loss = criterion(logits, labels) 

        loss.backward() 
        net.float() ##
        optimizer.step()