bugfix pool_size

pull/2245/head
dong-hyun 2024-08-03 09:50:20 +09:00
parent 6fdc9d8243
commit ea33d6a74d
1 changed files with 1 additions and 1 deletions

View File

@ -431,7 +431,7 @@ default_cfgs = generate_default_cfgs({
hf_hub_id='naver-ai/rdnet_large.nv_in1k'),
'rdnet_large.nv_in1k_ft_in1k_384': _cfg(
hf_hub_id='naver-ai/rdnet_large.nv_in1k_ft_in1k_384',
input_size=(3, 384, 384), crop_pct=1.0),
input_size=(3, 384, 384), crop_pct=1.0, pool_size=(12, 12)),
})