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,
        save_config_kwargs={"overwrite": True},
    )


if __name__ == "__main__":
    main()