Skip to content

Commit

Permalink
fix: add docstring for image_size in Convnextv2 config (#33734)
Browse files Browse the repository at this point in the history
add docstring for image_size
  • Loading branch information
lucianosrp authored Sep 26, 2024
1 parent 6730485 commit e32521b
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@ class ConvNextV2Config(BackboneConfigMixin, PretrainedConfig):
The epsilon used by the layer normalization layers.
drop_path_rate (`float`, *optional*, defaults to 0.0):
The drop rate for stochastic depth.
image_size (`int`, *optional*, defaults to 224):
The size (resolution) of each image.
out_features (`List[str]`, *optional*):
If used as backbone, list of features to output. Can be any of `"stem"`, `"stage1"`, `"stage2"`, etc.
(depending on how many stages the model has). If unset and `out_indices` is set, will default to the
Expand Down

0 comments on commit e32521b

Please sign in to comment.