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)
