tabensemb.model.AbstractNN.freeze

method

AbstractNN.freeze() → None

Freeze all params for inference.

Example:

model = MyLightningModule(...)
model.freeze()