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?