mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

pool size test fixes

Browse Source
This commit is contained in:
Ross Wightman 2024-08-17 08:27:13 -07:00
parent 1bd92bca0e
commit 7d83749207
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientnet.py
View File

@ -1294,7 +1294,7 @@ default_cfgs = generate_default_cfgs({
'efficientnet_b1.ra4_e3600_r240_in1k': _cfg( 'efficientnet_b1.ra4_e3600_r240_in1k': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
input_size=(3, 240, 240), crop_pct=0.9, input_size=(3, 240, 240), crop_pct=0.9, pool_size=(8, 8),
test_input_size=(3, 288, 288), test_crop_pct=1.0), test_input_size=(3, 288, 288), test_crop_pct=1.0),
'efficientnet_b1.ft_in1k': _cfg( 'efficientnet_b1.ft_in1k': _cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b1-533bc792.pth', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b1-533bc792.pth',

1
timm/models/hieradet_sam2.py
View File

@ -294,6 +294,7 @@ class HieraDet(nn.Module):
assert len(stages) == len(window_spec) assert len(stages) == len(window_spec)
self.num_classes = num_classes self.num_classes = num_classes
self.window_spec = window_spec self.window_spec = window_spec
self.output_fmt = 'NHWC'
depth = sum(stages) depth = sum(stages)
self.q_stride = q_stride self.q_stride = q_stride