Skip to content

Commit e32521b

Browse files
authored
fix: add docstring for image_size in Convnextv2 config (#33734)
add docstring for image_size
1 parent 6730485 commit e32521b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/transformers/models/convnextv2/configuration_convnextv2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ class ConvNextV2Config(BackboneConfigMixin, PretrainedConfig):
5252
The epsilon used by the layer normalization layers.
5353
drop_path_rate (`float`, *optional*, defaults to 0.0):
5454
The drop rate for stochastic depth.
55+
image_size (`int`, *optional*, defaults to 224):
56+
The size (resolution) of each image.
5557
out_features (`List[str]`, *optional*):
5658
If used as backbone, list of features to output. Can be any of `"stem"`, `"stage1"`, `"stage2"`, etc.
5759
(depending on how many stages the model has). If unset and `out_indices` is set, will default to the

0 commit comments

Comments
 (0)