tabensemb.model.AbstractNN.to_uncompiled

method

classmethod AbstractNN.to_uncompiled(model: pl.LightningModule | torch._dynamo.OptimizedModule) → pl.LightningModule

Returns an instance of LightningModule without any compilation optimizations from a compiled model.

This takes either a torch._dynamo.OptimizedModule returned by torch.compile() or a LightningModule returned by LightningModule.from_compiled.

Note: this method will in-place modify the LightningModule that is passed in.