fetch_pretrained_weights¶

fetch_pretrained_weights(model_name, save_path=None, *, overwrite=False)[source]¶

Get the pretrained model information from yml file.

Parameters:

  • model_name (str) – Refer to ::py::meth:get_pretrained_model for all supported model names.

  • save_path (str | Path) – Path to save the weight of the corresponding model_name.

  • overwrite (bool) – Overwrite existing downloaded weights.

Returns:

The local path to the cached pretrained weights after downloading.

Return type:

Path