bitorch.models.model_hub.load_from_hub

bitorch.models.model_hub.load_from_hub(model_version_table_path: str, download_path: str = 'bitorch_models', **model_kwargs: str) → Tensor

loads the model that matches the requested model configuration in model_kwargs from the model hub.

Parameters:

• model_version_table_path (str) – path to model version table on model hub

• download_path (str, optional) – path to store the downloaded files. Defaults to “/tmp”.

Returns:

state dict of downloaded model file

Return type:

torch.Tensor