17 lines
299 B
Python
17 lines
299 B
Python
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()
|