Epoch: [1][2120/5005]   Loss 3.5405e+00 (3.6588e+00)    Acc@1  25.00 ( 25.53)   Acc@5  51.56 ( 49.25)
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
Epoch: [1][2130/5005]   Loss 3.7294e+00 (3.6584e+00)    Acc@1  23.05 ( 25.54)   Acc@5  50.39 ( 49.26)
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(True, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
Epoch: [1][2140/5005]   Loss nan (nan)  Acc@1   0.00 ( 25.49)   Acc@5   1.17 ( 49.18)
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
Epoch: [1][2150/5005]   Loss nan (nan)  Acc@1   0.00 ( 25.38)   Acc@5   0.78 ( 48.96)
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
Epoch: [1][2160/5005]   Loss nan (nan)  Acc@1   0.00 ( 25.26)   Acc@5   0.78 ( 48.73)
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')
tensor(True) tensor(True, device='cuda:0') tensor(False, device='cuda:0')

I got this message when add this code.