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

Missing stem_kernel_size argument in EfficientNetFeatures

Browse Source
This commit is contained in:
Ross Wightman 2024-06-14 13:39:31 -07:00
parent 9613c76844
commit a0bb5b4a44
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/efficientnet.py
View File

@ -79,8 +79,8 @@ class EfficientNet(nn.Module):
num_features: int = 1280, num_features: int = 1280,
in_chans: int = 3, in_chans: int = 3,
stem_size: int = 32, stem_size: int = 32,
fix_stem: bool = False,
stem_kernel_size: int = 3, stem_kernel_size: int = 3,
fix_stem: bool = False,
output_stride: int = 32, output_stride: int = 32,
pad_type: str = '', pad_type: str = '',
act_layer: Optional[LayerType] = None, act_layer: Optional[LayerType] = None,
@ -278,6 +278,7 @@ class EfficientNetFeatures(nn.Module):
feature_location: str = 'bottleneck', feature_location: str = 'bottleneck',
in_chans: int = 3, in_chans: int = 3,
stem_size: int = 32, stem_size: int = 32,
stem_kernel_size: int = 3,
fix_stem: bool = False, fix_stem: bool = False,
output_stride: int = 32, output_stride: int = 32,
pad_type: str = '', pad_type: str = '',
@ -300,7 +301,7 @@ class EfficientNetFeatures(nn.Module):
# Stem # Stem
if not fix_stem: if not fix_stem:
stem_size = round_chs_fn(stem_size) stem_size = round_chs_fn(stem_size)
self.conv_stem = create_conv2d(in_chans, stem_size, 3, stride=2, padding=pad_type) self.conv_stem = create_conv2d(in_chans, stem_size, stem_kernel_size, stride=2, padding=pad_type)
self.bn1 = norm_act_layer(stem_size, inplace=True) self.bn1 = norm_act_layer(stem_size, inplace=True)
# Middle stages (IR/ER/DS Blocks) # Middle stages (IR/ER/DS Blocks)