Pretrained model loading

[xuhang2019]

Many thanks for the contribution to FL community. Really benefit a lot.

When I wanted to load the pretrained model, I didn't find a universal/easy way to do it, (i.e., in args).

After checking the code, I believe I should change the first parameter in trainable_params change to my desired checkpoints.

        self.model = use_model.to(self.device)

        # FIXME: using pre-trained models
        init_trainable_params, self.trainable_params_name = trainable_params(
            self.model, detach=True, requires_name=True
        )

Am I corrected? Thank you very much for your reply!

[KarhouTam]

Thanks for your attention first, @xuhang2019 .

For your question, yeah, you code in the right place 😄. NOTE: If you wanna load your own pretrained parameters, you need to confirm the parameters' shape is identical to the model's by yourself.

Seems I developed the saving model parameters feature but forgot the loading custom model parameters part 😂.

That feature will be on my list.

[xuhang2019]

Many thanks. Look forward to it! Brilliant.