fetch_pretrained_weights

fetch_pretrained_weights(model_name, save_path, overwrite=True)[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 to save the weight of the corresponding model_name.

  • overwrite (bool) – Overwrite existing downloaded weights.