MultiModalPredictor.dump_model

MultiModalPredictor.dump_model(save_path: str | None = None)[源]

将模型权重和配置保存到本地目录。模型权重保存在文件 pytorch_model.bin 中(适用于 timm_imagehf_text),或保存在 ‘<ckpt_name>.pth’ 中(适用于 mmdet_image)。配置保存在文件 config.json 中(适用于 timm_imagehf_text),或保存在 ‘<ckpt_name>.py’ 中(适用于 mmdet_image)。

参数:

save_path (str) – 保存模型和配置的目录路径。