InternImage icon indicating copy to clipboard operation
InternImage copied to clipboard

Published 20 hours ago • verified publisher icon OpenGVLab

[bug] how to train fine-tuning classification model (size mismatch for head.bias: copying a param with shape torch.Size([1000]) from checkpoint, the shape in current model is torch.Size([4]).)

Open jeonga0303 opened this issue 8 months ago • 3 comments

image

I customized the config.py. how to train fine-tuning classification model?

image

jeonga0303 avatar Jun 05 '24 06:06 jeonga0303