KeyError: 'state_dict' running test.py

[danielzhangau]

trafficstars

After I trained the model with only drivable area detection task, I wanna evaluate my model by using test.py It raised me the following key error

[danielzhangau]

I avoid the issue by using 'final_state.pth'. instead, I used 'epoch_240.pth'. Don't know why but I am running now.

[szh-1598]

我也有同样的问题，请问你是怎样解决的？

[yaoshanliang]

同问