for both BCEWithLogitsLoss and CrossEntropyLoss ( 1 step )
we will need to do this when doing inferencing?
logps = model(img)
ps = torch.exp(logps)

Also, even if it’s 2steps(i.e logsoftmax + nlllosss) the above still applies right?
Thanks