finetune能不能加个功能

[404289680]

Traceback (most recent call last):
File "./fastreid/engine/train_loop.py", line 125, in train
self.run_step()
File "./fastreid/engine/train_loop.py", line 244, in run_step
self.optimizer.step()
File "/home/zuwei/anaconda3/lib/python3.8/site-packages/torch/optim/lr_scheduler.py", line 67, in wrapper
return wrapped(*args, **kwargs)
File "/home/zuwei/anaconda3/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 15, in decorate_context
return func(*args, **kwargs)
File "./fastreid/solver/optim/adam.py", line 102, in step
exp_avg.mul_(beta1).add_(grad, alpha=1 - beta1)
RuntimeError: The size of tensor a (50000) must match the size of tensor b (25754) at non-singleton dimension 0
类别数不一样，无法finetune

[L1aoXingyu]

fixed.
You can finetune from the other trained model by calling

python3 tools/train_net.py --config-file CONFIG_FILE_PATH --num-gpus 2 MODEL.WEIGHTS PRETRAIN_MODEL_PATH

[DeepAlchemist]

If the current model is partially different from the trained model by adding some parameters, it raises errors.

BTW, I can not figure out the differences between --finetune, resume=True, and resume=False.