@shivam2298 Hey i worked.Thank You.
Please rectify what mistake i was doing(I am not fully understood).(i have to convert the prediction again back to image)
And my training loss is also decreasing to slowly.
criterion =  nn.CrossEntropyLoss()
# specify optimizer
optimizer = optim.Adam(model.parameters(), lr=1e-5,eps=1e-08)

Any help u can provide with this.