Thank you for your recommendation about Hydra, that looks interesting and potentially like a great help. I also came across https://lightning.ai/forums/t/best-way-to-use-load-from-checkpoint-when-model-contains-other-models/2094/2 so going with option A would eliminate the need to pass additional arguments during losing the checkpoint and one could pass the model class name as a variable as well to keep the flexibility from using different backbone models.
1 Like