MengyuWang826 / SegRefiner

SegRefiner: Towards Model-Agnostic Segmentation Refinement with Discrete Diffusion Process
Apache License 2.0
137 stars 8 forks source link

KeyError:'meta' #9

Open wzx0720 opened 5 months ago

wzx0720 commented 5 months ago

作者您好!感谢您的工作! 我想在我自己的数据集上训练HR-SegRefiner,当我尝试使用您提供的训练好的segrefiner_hr_latest.pth作为预训练模型时,我使用命令python tools/train.py configs/segrefiner/segrefiner_hr.py --resume-from segrefiner_hr_latest.pth却出现了如下错误: self._epoch = checkpoint['meta']['epoch'] KeyError:'meta'

但是这一权重模型在进行测试时并没有报错。请问我应该如何解决这一问题?感谢您的回复!