Skip to content

Commit

Permalink
Update src/transformers/models/vitpose/configuration_vitpose.py
Browse files Browse the repository at this point in the history
  • Loading branch information
SangbumChoi authored Sep 6, 2024
1 parent cbbb966 commit f0e0f79
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/models/vitpose/configuration_vitpose.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ class ViTPoseConfig(PretrainedConfig):
scale_factor (`int`, *optional*, defaults to 4):
Factor to upscale the feature maps coming from the ViT backbone.
use_simple_decoder (`bool`, *optional*, defaults to `True`):
Whether to use a simple decoder to decode the feature maps from the backbone into heatmaps.
Whether to use a `VitPoseSimpleDecoder` to decode the feature maps from the backbone into heatmaps. Otherwise it uses `VitPoseClassicDecoder`.
Example:
Expand Down

0 comments on commit f0e0f79

Please sign in to comment.