I load my own model (torch) from pretrained path, while I could not freeze them.
self.model = Model.from_pretrained("some_model") self.model.freeze()
I have error :
AttributeError: 'Model' object has no attribute 'freeze'.
I guess because the model is not LightningModule.
If the model is not LightningModule, how to freeze this model?