Solved by Balamurali_M in post #4 
              
              
                I think running_corrects will be an integer and the denominator will also be an integer. First the number should be converted to float and perform division. And the tensor should be printed with cpu().numpy() only. Which version of torch are you using ? Can you provide more inputs if this does’nt w…