Merge pull request #42 from yoshitomo-matsubara/dev

Fix bugs
yoshitomo-matsubara · Dec 27, 2020 · ca2d86d · ca2d86d
2 parents ebca59e + 42943f6
commit ca2d86d
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
               'Trained models, training logs and configurations are available for ensuring the reproducibiliy.'
 setup(
     name='torchdistill',
-    version='0.1.0',
+    version='0.1.1',
     description=description,
     long_description=long_description,
     long_description_content_type='text/markdown',

diff --git a/torchdistill/models/classification/densenet.py b/torchdistill/models/classification/densenet.py
@@ -16,8 +16,8 @@
 
 ROOT_URL = 'https://github.com/yoshitomo-matsubara/torchdistill/releases/download'
 MODEL_URL_DICT = {
-    'cifar10-densenet_bc_k12_depth100': ROOT_URL + '/v0.1.0/cifar10-densenet_bc_k12_depth100.pt',
-    'cifar100-densenet_bc_k12_depth100': ROOT_URL + '/v0.1.0/cifar100-densenet_bc_k12_depth100.pt'
+    'cifar10-densenet_bc_k12_depth100': ROOT_URL + '/v0.1.1/cifar10-densenet_bc_k12_depth100.pt',
+    'cifar100-densenet_bc_k12_depth100': ROOT_URL + '/v0.1.1/cifar100-densenet_bc_k12_depth100.pt'
 }
 
 

diff --git a/torchdistill/models/classification/resnet.py b/torchdistill/models/classification/resnet.py
@@ -14,11 +14,11 @@
 
 ROOT_URL = 'https://github.com/yoshitomo-matsubara/torchdistill/releases/download'
 MODEL_URL_DICT = {
-    'cifar10-resnet20': ROOT_URL + '/v0.1.0/cifar10-resnet20.pt',
-    'cifar10-resnet32': ROOT_URL + '/v0.1.0/cifar10-resnet32.pt',
-    'cifar10-resnet44': ROOT_URL + '/v0.1.0/cifar10-resnet44.pt',
-    'cifar10-resnet56': ROOT_URL + '/v0.1.0/cifar10-resnet56.pt',
-    'cifar10-resnet110': ROOT_URL + '/v0.1.0/cifar10-resnet110.pt'
+    'cifar10-resnet20': ROOT_URL + '/v0.1.1/cifar10-resnet20.pt',
+    'cifar10-resnet32': ROOT_URL + '/v0.1.1/cifar10-resnet32.pt',
+    'cifar10-resnet44': ROOT_URL + '/v0.1.1/cifar10-resnet44.pt',
+    'cifar10-resnet56': ROOT_URL + '/v0.1.1/cifar10-resnet56.pt',
+    'cifar10-resnet110': ROOT_URL + '/v0.1.1/cifar10-resnet110.pt'
 }
 
 

diff --git a/torchdistill/models/classification/wide_resnet.py b/torchdistill/models/classification/wide_resnet.py
@@ -13,12 +13,12 @@
 
 ROOT_URL = 'https://github.com/yoshitomo-matsubara/torchdistill/releases/download'
 MODEL_URL_DICT = {
-    'cifar10-wide_resnet40_4': ROOT_URL + '/v0.1.0/cifar10-wide_resnet40_4.pt',
-    'cifar10-wide_resnet28_10': ROOT_URL + '/v0.1.0/cifar10-wide_resnet28_10.pt',
-    'cifar10-wide_resnet16_8': ROOT_URL + '/v0.1.0/cifar10-wide_resnet16_8.pt',
-    'cifar100-wide_resnet40_4': ROOT_URL + '/v0.1.0/cifar100-wide_resnet40_4.pt',
-    'cifar100-wide_resnet28_10': ROOT_URL + '/v0.1.0/cifar100-wide_resnet28_10.pt',
-    'cifar100-wide_resnet16_8': ROOT_URL + '/v0.1.0/cifar100-wide_resnet16_8.pt'
+    'cifar10-wide_resnet40_4': ROOT_URL + '/v0.1.1/cifar10-wide_resnet40_4.pt',
+    'cifar10-wide_resnet28_10': ROOT_URL + '/v0.1.1/cifar10-wide_resnet28_10.pt',
+    'cifar10-wide_resnet16_8': ROOT_URL + '/v0.1.1/cifar10-wide_resnet16_8.pt',
+    'cifar100-wide_resnet40_4': ROOT_URL + '/v0.1.1/cifar100-wide_resnet40_4.pt',
+    'cifar100-wide_resnet28_10': ROOT_URL + '/v0.1.1/cifar100-wide_resnet28_10.pt',
+    'cifar100-wide_resnet16_8': ROOT_URL + '/v0.1.1/cifar100-wide_resnet16_8.pt'
 }