I am training a model with Pytorch lightning. I save .ckpt checkpoints. But they never get uploaded to clearml! How do I make clearml detect the checkpoints?
I tried:
task = Task.init(
project_name=args.project,
task_name=f"Train {args.model_name}",
task_type=TaskTypes.training,
output_uri=True,
auto_connect_frameworks={
"pytorch": ["*.pt", "*.pth", "*.ckpt"],
"lightning": ["*.ckpt"],
"pytorch_lightning": ["*.ckpt"],
"pytorch-lightning": ["*.ckpt"],
"pytorch lightning": ["*.ckpt"],
},
)
But it doesnt work
AdventurousButterfly15 , to upload artifacts automatically you can use the output_uri parameter of Task.init()
None
Looking through history I found this link: None
Tldr: ClearML doesn’t support lightning , but supports pytorch_lightning . Downgrading from the new interface to the old one fixed my issue
Hi AdventurousButterfly15 , are the models logged in the artifacts section?
