predict_nrrd.py do error

[wangchenguang123]

when i finshed training this model,i want to do some predict. but when i do this py ,i find i canot get the pred
pred = outputs.data.max(1)[1].cpu().numpy() this code result show is empty ,so i want to ask why about it, And how can i do to solve this problem
Thank you

[wangchenguang123]

@cherise215

[cherise215]

Can you check your input dim? thanks.