How to launch a separate fit within fit? #11063
Unanswered
RylanSchaeffer
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I want to train a network, and then after every K (a hyperparameter) epochs, attach a linear layer to the network and finetune the linear layer on a separate task (i.e. train and test using different dataloaders) for some number of epochs before returning to the main training loop. Is there any easy way to do this within PyTorch Lightning? The key challenge is doing this every K epochs, not before/after all epochs or before/after every epoch.
Beta Was this translation helpful? Give feedback.
All reactions