13 lines
228 B
Python
13 lines
228 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)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|