How to load pretrain models? #15597
xiazhongyv
started this conversation in
General
Replies: 1 comment
-
I tried using
|
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I have a lightning model {EncoderA, EncoderB, FusionModule, Decoder}
And I pretrained (DDP ckpt1){EncoderA, Decoder}, (DDP ckpt2){EncoderB, Decoder}
How can I train (DDP){EncoderA, EncoderB, FusionModule, Decoder}, load pretrained EncoderA from ckpt1, load pretrained EncoderB from ckpt2, and ignore hyperparameters of ckpt1 and ckpt2 ?
Beta Was this translation helpful? Give feedback.
All reactions