mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

🎨 Import load_state_dict_from_url directly

pull/842/head
nateraw 2021-09-14 01:11:40 -04:00
parent e65a2cba3d
commit adcb74f87f
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/helpers.py
View File

@ -11,10 +11,10 @@ from typing import Any, Callable, Optional, Tuple
import torch
import torch.nn as nn
from torch.hub import load_state_dict_from_url
from .features import FeatureListNet, FeatureDictNet, FeatureHookNet
from .hub import has_hf_hub, download_cached_file, load_state_dict_from_hf, load_state_dict_from_url
from .hub import has_hf_hub, download_cached_file, load_state_dict_from_hf
from .layers import Conv2dSame, Linear

2
timm/models/hub.py
View File

@ -6,7 +6,7 @@ from pathlib import Path
from typing import Union
import torch
from torch.hub import HASH_REGEX, download_url_to_file, urlparse, load_state_dict_from_url
from torch.hub import HASH_REGEX, download_url_to_file, urlparse
try:
from torch.hub import get_dir
except ImportError: