Skip to content

Commit 6bcbdbf

Browse files
committed
CS3-DarkNet Small (Focus) w/ RA4 recipe. Fix #2122
1 parent 3a6cc4f commit 6bcbdbf

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

timm/models/cspnet.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -963,7 +963,10 @@ def _cfg(url='', **kwargs):
963963
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_x_c2ns-4e4490aa.pth',
964964
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),
965965

966-
'cs3darknet_focus_s.untrained': _cfg(interpolation='bicubic'),
966+
'cs3darknet_focus_s.ra4_e3600_r256_in1k': _cfg(
967+
hf_hub_id='timm/',
968+
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),
969+
interpolation='bicubic', test_input_size=(3, 320, 320), test_crop_pct=1.0),
967970
'cs3darknet_focus_m.c2ns_in1k': _cfg(
968971
hf_hub_id='timm/',
969972
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tpu-weights/cs3darknet_focus_m_c2ns-e23bed41.pth',

0 commit comments

Comments
 (0)