Haoran-SONG / PiP-Planning-informed-Prediction

(ECCV 2020) PiP: Planning-informed Trajectory Prediction for Autonomous Driving
http://song-haoran.com/planning-informed-prediction/
176 stars 45 forks source link

报错问题 #13

Open Yuchao-png opened 2 years ago

Yuchao-png commented 2 years ago

您好,我有一个问题,当我关闭planning模块时,evaluate会报错:RuntimeError: Error(s) in loading state_dict for pipNet: Unexpected key(s) in state_dict: "plan_lstm.weight_ih_l0", "plan_lstm.weight_hh_l0", "plan_lstm.bias_ih_l0", "plan_lstm.bias_hh_l0", "plan_conv_social.0.weight", "plan_conv_soc ial.0.bias", "plan_conv_social.3.weight", "plan_conv_social.3.bias". 我不知道如何解决,希望您百忙之中可以解答我的问题

YongLi000 commented 7 months ago

Traceback (most recent call last): File "evaluate.py", line 182, in model_evaluate() …………………………………………………… File "/home/robot/.virtualenvs/traj_prediction_v1/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 201, in forward return F.conv1d(input, self.weight, self.bias, self.stride, RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR

上面是报错信息,我的cuda是11.5,与我的cuDNN8.x是兼容的,但是仍然报错,请问是什么情况呢?非常感谢您的,期待您的回复