Traceback (most recent call last):
File "/app/train_semi.py", line 931, in
main(opt)
File "/app/train_semi.py", line 829, in main
train(opt.hyp, opt, device, callbacks)
File "/app/train_semi.py", line 510, in train
_update_teacher_model(model if ema is None else ema.ema, model_teacher, word_size=WORLD_SIZE, keep_rate=0.)
File "/opt/conda/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/app/utils/torch_utils.py", line 48, in _update_teacher_model
raise Exception("{} is not found in student model".format(key))
Exception: model.0.conv.conv.weight is not found in student model
after resolving all problems about training my custom dataset with One teacher when I launch the training, I got this error:
Traceback (most recent call last): File "/app/train_semi.py", line 931, in
main(opt)
File "/app/train_semi.py", line 829, in main
train(opt.hyp, opt, device, callbacks)
File "/app/train_semi.py", line 510, in train
_update_teacher_model(model if ema is None else ema.ema, model_teacher, word_size=WORLD_SIZE, keep_rate=0.)
File "/opt/conda/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/app/utils/torch_utils.py", line 48, in _update_teacher_model
raise Exception("{} is not found in student model".format(key))
Exception: model.0.conv.conv.weight is not found in student model
How could I resolve it ?