Thanks, but what will be a ideal Train loss to be? I am doing it in 30 epoch and it is from 1.35, to 0.83 after 30 epoch. Is it okay? or should I do more epoch or make learning rate more small? This is my model -
model = models.resnet101(pretrained=True)

for param in model.parameters():
    param.requires_grad = True

model.fc = nn.Sequential(nn.Linear(2048, 1024),
                                 nn.ReLU(),
                                 nn.Dropout(0.4),
                                 nn.Linear(1024, 512),
                                 nn.ReLU(),
                                 nn.Dropout(0.4),
                                 nn.Linear(512,4),
                                 nn.LogSoftmax(dim=1))

criterion = nn.NLLLoss()

optimizer = optim.Adam(model.fc.parameters(), lr = 0.000005)