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 the directory in which the pretrained weight will be cached.

  • overwrite (bool) – Overwrite existing downloaded weights (force downloading).

Returns:

The local path to the cached pretrained weights after downloading.

Return type:

Path