ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

MLflow 3.x 用 search_logged_models 列出未注册的已记录模型

MLflow 3.x 用 search_logged_models 列出未注册的已记录模型

训练过程中每轮把更优的 checkpoint 用mlflow.pytorch.log_model记下来,并且不注册到 Model Registry,事后想用代码把这些模型枚举出来做推理——这一步在 MLflow 3.x 里和旧的 artifact 思路不一样,用list_artifacts会一无所获。

问题现象

记录模型的写法:

mlflow.pytorch.log_model(model_cpu,name=name,# 例如 checkpoint_epoch_2step=epoch,signature=self
RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进