torch_hub_load_github#

torch_hub_load_github(repo: str, entrypoint: str, *args, dst_dir: Optional[str] = None, **kwargs) Any[source]#

Load an entrypoint from a github repo using torch.hub.load().

Parameters
  • repo (str) – <repo-owner>/<erpo-name>[:tag]

  • entrypoint (str) – Name of a callable present in hubconf.py.

  • *args – Args to be passed to the entrypoint.

  • dst_dir (Optional[str]) – If provided, the contents of the repo are copied there. Defaults to None.

  • **kwargs – Keyword args to be passed to the entrypoint.

Returns

The output from calling the entrypoint.

Return type

Any