I’m getting this…
Epoch 18 		 Training Loss: 0.004407357424497604 		 Validation Loss: 8.263123961403024e-08
Validation Accuracy … : tensor(0.4572, device=‘cuda:0’)
Train Accuracy … : tensor(0.9526, device=‘cuda:0’)
I’m using optimizer = optim.SGD(net.parameters(), lr = 0.0001,momentum = 0.92) and
net.fc = nn.Sequential(nn.Linear(2048,512,bias = True),

                               nn.ReLU(inplace=True),

                               nn.Linear(512,6),

                      )

in Resnet50. @eqy