Traceback (most recent call last):


 "train.py", line 232, in <module>

main()


 "train.py", line 152, in main

for _, train_data in enumerate(train_loader):


 "/home/pjh2/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 582, in __next__

return self._process_next_batch(batch)


 "/home/pjh2/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):

Traceback (most recent call last):


 "eval_voc.py", line 113, in <module>

visiual=opt.visiual).detection()


 "eval_voc.py", line 34, in __init__

self.__load_model_weights(weight_path)


 "eval_voc.py", line 43, in __load_model_weights

self.__model.load_state_dict(chkpt)


 "C:\Anaconda3\envs\YOLOv4-pytorch\lib\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 Build_Model: