import torch
from lightning.pytorch.cli import LightningCLI

torch.set_float32_matmul_precision("medium")


def main():
    LightningCLI(subclass_mode_data=True, subclass_mode_model=True)


if __name__ == "__main__":
    main()