2023-05-11 12:59:11 +00:00
|
|
|
try:
|
|
|
|
|
from lightning.pytorch.cli import LightningCLI
|
|
|
|
|
from lightning.pytorch.demos.boring_classes import DemoModel, BoringDataModule
|
|
|
|
|
except ImportError:
|
|
|
|
|
import sys
|
|
|
|
|
print("Module 'lightning' not installed (only available for Python 3.8+")
|
|
|
|
|
sys.exit(0)
|
2022-03-29 08:00:33 +00:00
|
|
|
from clearml import Task
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2023-05-11 12:59:11 +00:00
|
|
|
Task.add_requirements("requirements.txt")
|
|
|
|
|
Task.init(project_name="example", task_name="pytorch_lightning_jsonargparse")
|
|
|
|
|
LightningCLI(DemoModel, BoringDataModule)
|
