Unanswered
Hello Everyone,
I'D Like To Stop Clearml From Automatically Logging Models From Yolov8. Is There Any Way To Do It Without Disconnecting From Framework (Pytorch)?
Hey 🙂 I had a similar issue today and found this solution:
In my case this codebase was using a .pt
filetype which was being picked up and logged as a model even though it was not.
import os
from clearml import Task
from clearml.binding.frameworks import WeightsFileHandler
task = Task.init(
project_name="task_project",
task_name="task_name",
task_type=Task.TaskTypes.training,
)
def filter_out_pt_files(operation_type, model_info):
is_pt_file = os.path.splitext(model_info.local_model_path)[-1] == ".pt"
if is_pt_file:
return None
return model_info
WeightsFileHandler.add_pre_callback(filter_out_pt_files)
...
158 Views
0
Answers
one year ago
one year ago