您好,欢迎访问这里是您的网站名称官网!
TAG标签

主页 > TAG标签 > 梯度

用自己训练的权重训练其他模型出现的问题ValueError: loaded state dict contains a parameter group that doesn‘t match the s2024-04-15
所以,在读取优化器记录的时候要注意,如果保存的数据是针对某些层保存的,而加载的时候又希望对所有层进行训练,这时候优化器数据就没必要加载了,要么写个try,要么就把这行注释掉。在模型中如果设置了requires_grad=True,则表示该层...

平台注册入口