Cogito2012 / UString

[ACM MM 2020] Uncertainty-based Traffic Accident Anticipation
MIT License
63 stars 18 forks source link

DAD model weight can't load #29

Open sumitmishra209 opened 10 months ago

sumitmishra209 commented 10 months ago

Hello, I am trying to load models and reproduce result. Crash weight get loaded and I got same result of paper but DAD model can't get load and some weights of layers are missing can you check and provide weights. Thanks!

self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for UString: Missing key(s) in state_dict: "enc_gcn1.weight", "enc_gcn2.weight". Unexpected key(s) in state_dict: "enc_layer1.weight", "enc_layer2.weight".