Open dudiaoxifenglie opened 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的列表,请问如何解决?
你好,关于这部分你需要新建一个长度为4的parameter groups并进行赋值.
好的,非常感谢
您好,首先非常感谢您的开源代码。我想在您的model_general.pth的基础上,进行自己数据集的训练,但是在状态加载的过程中,出现了如下问题: ValueError: loaded state dict has a different number of parameter groups。 初步查看是因为:model_general.pth 中保留的 parameter groups 为一个长度为3的列表,而新建立的optimiter 为一个长度为4的列表,请问如何解决?