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