StreamYOLO icon indicating copy to clipboard operation
StreamYOLO copied to clipboard

Published 20 hours ago • verified publisher icon yancie-yjr

KeyError: 'model'

Open HarryBarry123 opened this issue 1 year ago • 0 comments

when I tried to train File "/home/pe/projects/czy/StreamYOLO-main/exps/train_utils/double_trainer.py", line 314, in resume_train ckpt = torch.load(ckpt_file, map_location=self.device)["model"] │ │ │ │ └ 'cuda:0' │ │ │ └ <exps.train_utils.double_trainer.Trainer object at 0x7fe2e69a3650> │ │ └ '/home/pe/projects/czy/StreamYOLO-main/tools/yolox_s.pth' │ └ <function load at 0x7fe2e8dc8710> └ <module 'torch' from '/home/pe/anaconda3/envs/streamyolo/lib/python3.7/site-packages/torch/init.py'> KeyError: 'model' Could you tell me how to solve this?

HarryBarry123 avatar Nov 01 '23 03:11 HarryBarry123