ohhhyeahhh / SiamCAR

SiamCAR: Siamese Fully Convolutional Classification and Regression for Visual Tracking (CVPR 2020, Oral)
https://openaccess.thecvf.com/content_CVPR_2020/html/Guo_SiamCAR_Siamese_Fully_Convolutional_Classification_and_Regression_for_Visual_Tracking_CVPR_2020_paper.html
323 stars 63 forks source link

使用model_general.pth进行在训练的问题 #64

Open dudiaoxifenglie opened 3 years ago

dudiaoxifenglie commented 3 years ago

您好,首先非常感谢您的开源代码。我想在您的model_general.pth的基础上,进行自己数据集的训练,但是在状态加载的过程中,出现了如下问题: ValueError: loaded state dict has a different number of parameter groups。 初步查看是因为:model_general.pth 中保留的 parameter groups 为一个长度为3的列表,而新建立的optimiter 为一个长度为4的列表,请问如何解决?

WangJun-ZJUT commented 3 years ago

你好,关于这部分你需要新建一个长度为4的parameter groups并进行赋值.

dudiaoxifenglie commented 3 years ago

好的,非常感谢