Manually call model.eval() and model.train() inside the training loop
How to manually call model.eval() or model.train() inside the lightning module? I happen to have several models and not all of them need to be updated during each forward pass. Thanks!
You can use self.freeze() and self.unfreeze().