excuse me, may I ask you a couple of questions?

harryhan618 / SCNN_Pytorch

Pytorch implementation of "Spatial As Deep: Spatial CNN for Traffic Scene Understanding"

MIT License

250 stars 68 forks source link

excuse me, may I ask you a couple of questions? #56

Closed HejhGitHub closed 3 years ago

HejhGitHub commented 3 years ago

For single image demo test,I used the exp0_best.pth to test the single image,but it show the next errors: RuntimeError: Error(s) in loading state_dict for SCNN: size mismatch for fc.0.weight: copying a param with shape torch.Size([128, 2880]) from checkpoint, the shape in current model is torch.Size([128, 4500]). How to work out ?

harryhan618 commented 3 years ago

Are you asking the same question as in issue #54 ?