Once I fine-tune a model, I’d like to perform experiments with different heads but the same well-trained body.
To this end, I’d like to save just the body’s weights, and then instantiate a learner passing those weights like a pretrained model, while passing a head Kaimig-initialized.