Skip to content

Commit 72f0edb

Browse files
committed
missed first_conv for rnv2 18d
1 parent 3ed603a commit 72f0edb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/resnetv2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -699,7 +699,7 @@ def _cfg(url='', **kwargs):
699699
'resnetv2_18.untrained': _cfg(
700700
interpolation='bicubic', crop_pct=0.95),
701701
'resnetv2_18d.untrained': _cfg(
702-
interpolation='bicubic', crop_pct=0.95),
702+
interpolation='bicubic', crop_pct=0.95, first_conv='stem.conv1'),
703703
'resnetv2_50.a1h_in1k': _cfg(
704704
hf_hub_id='timm/',
705705
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),

0 commit comments

Comments
 (0)