def test_load_from_local_dir(self): local_dir = hub._get_cache_or_reload('ailzhang/torchhub_example', force_reload=False) hub_model = hub.load(local_dir, 'mnist', source='local', pretrained=True, verbose=False) self.assertEqual(sum_of_state_dict(hub_model.state_dict()), SUM_OF_HUB_EXAMPLE)
def repo(github, force_reload=False, verbose=True): # pytorch-1.6.0 applicable from torch.hub import _get_cache_or_reload import os if True: os.makedirs(get_dir(), exist_ok=True) else: hubdir = f"{_get_ml_home()}/hub" os.makedirs(hubdir, exist_ok=True) set_dir(hubdir) return _get_cache_or_reload(github, force_reload, verbose)
def reload_github(github): hub._get_cache_or_reload(github, force_reload=True, verbose=False)