yeah I  am using torch.max(output,1) for getting class prediction,  I have evaluated for 5000 images , accuracy of prediction is same as what i was getting before(from base code)
but my concern is can output be negative? if it soo I can move further