Closed 1145284121 closed 2 years ago
Scaled weight_decay = 0.0004921875 Optimizer groups: 135 .bias, 148 conv.weight, 121 other Traceback (most recent call last): File "train_omc.py", line 742, in main(opt) File "train_omc.py", line 640, in main train(opt.hyp, opt, device) File "train_omc.py", line 266, in train optimizer.load_state_dict(ckpt['optimizer']) File "/root/miniconda3/envs/omc/lib/python3.8/site-packages/torch/optim/optimizer.py", line 146, in load_state_dict raise ValueError("loaded state dict contains a parameter group " ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group
训练使用单卡2080ti,batch 6
如果跳过optimizer的话,载入siammot也会报错
我也遇到了这个问题
same problem...
Scaled weight_decay = 0.0004921875 Optimizer groups: 135 .bias, 148 conv.weight, 121 other Traceback (most recent call last): File "train_omc.py", line 742, in
main(opt)
File "train_omc.py", line 640, in main
train(opt.hyp, opt, device)
File "train_omc.py", line 266, in train
optimizer.load_state_dict(ckpt['optimizer'])
File "/root/miniconda3/envs/omc/lib/python3.8/site-packages/torch/optim/optimizer.py", line 146, in load_state_dict
raise ValueError("loaded state dict contains a parameter group "
ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group
训练使用单卡2080ti,batch 6
如果跳过optimizer的话,载入siammot也会报错