error in script predict_from_net.py

[Ray Zhou]

In ./shapenet/scripts/predict_from_net.py line 118
pred = net(img_tensor).cpu().numpy()["pred"][0]
should be
pred = net(img_tensor).cpu().numpy()[0]

obviously net(img_tensor).cpu().numpy() is not a dict

[Justus Schock]

Which delira version are you using? Because with the current version it should be a dict

[Ray Zhou]

I'm using 0.4.1
After removing ["pred"] it works fine

[Justus Schock]

That's strange. It should be a dict. I don't know, when I have the time to look into this, but I'll try to do so asap