Open shy-7 opened 1 year ago
trainer/ssod_trainer.py 大概144行处
if self.ema and ckpt.get('ema'):
self.ema.ema.load_state_dict(ckpt['ema'].float().state_dict(), strict=False)
改为:
if self.ema and ckpt.get('ema'):
csd = intersect_dicts(ckpt['ema'].float().state_dict(), self.model.state_dict(), exclude=exclude) #---加的就是这行
self.ema.ema.load_state_dict(csd, strict=False)
还是报错啊
你用的pt是不是在coco上面训练的,这个pt应该也要自己训练吧