Skip to content

Commit 4042a94

Browse files
committed
Add weights for two 'Edge' block (3x3->1x1) variants of CS3 networks.
1 parent c8f69e0 commit 4042a94

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

timm/models/cspnet.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,9 +97,11 @@ def _cfg(url='', **kwargs):
9797
url='', interpolation='bicubic'),
9898

9999
'cs3edgenet_x': _cfg(
100-
url='', interpolation='bicubic'),
100+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3edgenet_x_c2-2e1610a9.pth',
101+
interpolation='bicubic', test_input_size=(3, 288, 288), test_crop_pct=1.0),
101102
'cs3se_edgenet_x': _cfg(
102-
url='', interpolation='bicubic'),
103+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3se_edgenet_x_c2ns-76f8e3ac.pth',
104+
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0),
103105
}
104106

105107

0 commit comments

Comments
 (0)