CLI with torch.compile()

[Hannibal046]

Hi, how to integrate torch.compile() with CLI ?

class Model(L.LightningModule):
    def __init__(self) -> None:
        self.model = construct_model()
        self.model = torch.compile(self.model)

Is this the recommended way?