MultiModalPredictor.dump_model

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

Save model weights and config to a local directory. Model weights are saved in the file pytorch_model.bin (for timm_image or hf_text) or ‘<ckpt_name>.pth’ (for mmdet_image). Configs are saved in the file config.json (for timm_image or hf_text) or ‘<ckpt_name>.py’ (for mmdet_image).

Parameters:

save_path (str) – Path to directory where models and configs should be saved.