the models errors

[oxm]

Traceback (most recent call last):
File "submission_kitti.py", line 83, in
model.load_state_dict(state_dict['state_dict'], strict=True)
File "/home/xxm/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 845, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DeepPruner:

[ShivamDuggal4]

Hi @oxm , is this the entire error message? If not, could you please post the entire error message.

Potentially, the error could be because the 'models/config.py' hasn't been set according to the type of the model and the dataset you want to run.

Please check 'models/config.py' and set it according to your use-case.
'cost_aggregator_scale' should be 4 for DeepPruner-best-* and 8 for DeepPruner-fast-*

[oxm]

yes，you are right