Testing Script: Test On the custom Training

[NaeemKhan333]

I have trained my model on the custom data set. The model get trained and checkpoints weights get save in the folder . Now I am trying to run test script using the saved checkpoints it is giving me following error
Testing command

python3 test_video.py --arch BNInception --dataset something --weights pretrain/TRN_something_RGB_BNInception_TRN_segment1_checkpoint.pth.tar --frame_folder sample_data/vid

Multi-Scale Temporal Relation Network Module in use ['8-frame relation', '7-frame relation', '6-frame relation', '5-frame relation', '4-frame relation', '3-frame relation', '2-frame relation']
Traceback (most recent call last):
File "test_video.py", line 111, in
net.load_state_dict(base_dict)
File "/home/naeem/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 839, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for TSN:
Missing key(s) in state_dict: "consensus.fc_fusion_scales.0.1.weight", "consensus.fc_fusion_scales.0.1.bias", "consensus.fc_fusion_scales.0.3.weight", "consensus.fc_fusion_scales.0.3.bias", "consensus.fc_fusion_scales.1.1.weight", "consensus.fc_fusion_scales.1.1.bias", "consensus.fc_fusion_scales.1.3.weight", "consensus.fc_fusion_scales.1.3.bias", "consensus.fc_fusion_scales.2.1.weight", "consensus.fc_fusion_scales.2.1.bias", "consensus.fc_fusion_scales.2.3.weight", "consensus.fc_fusion_scales.2.3.bias", "consensus.fc_fusion_scales.3.1.weight", "consensus.fc_fusion_scales.3.1.bias", "consensus.fc_fusion_scales.3.3.weight", "consensus.fc_fusion_scales.3.3.bias", "consensus.fc_fusion_scales.4.1.weight", "consensus.fc_fusion_scales.4.1.bias", "consensus.fc_fusion_scales.4.3.weight", "consensus.fc_fusion_scales.4.3.bias", "consensus.fc_fusion_scales.5.1.weight", "consensus.fc_fusion_scales.5.1.bias", "consensus.fc_fusion_scales.5.3.weight", "consensus.fc_fusion_scales.5.3.bias", "consensus.fc_fusion_scales.6.1.weight", "consensus.fc_fusion_scales.6.1.bias", "consensus.fc_fusion_scales.6.3.weight", "consensus.fc_fusion_scales.6.3.bias".
Unexpected key(s) in state_dict: "consensus.classifier.1.weight", "consensus.classifier.1.bias", "consensus.classifier.3.weight", "consensus.classifier.3.bias".