MultiModalPredictor.dump_model

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

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

Parameters:

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