Merge branch 'master' into lrfind-fix

frgfm · Nov 1, 2021 · b6f6678 · b6f6678
2 parents 58d89f6 + db8c5fe
commit b6f6678
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/holocron/models/utils.py b/holocron/models/utils.py
@@ -8,7 +8,7 @@
 import torch.nn as nn
 from holocron.nn import BlurPool2d
 from typing import List, Optional, Any, Callable, Tuple
-from torchvision.models.utils import load_state_dict_from_url
+from torch.hub import load_state_dict_from_url
 
 
 __all__ = ['conv_sequence', 'load_pretrained_params', 'fuse_conv_bn']