Skip to content

Commit

Permalink
Merge branch 'fix_vit_pos_embed' of https://github.com/xiexinch/mmseg…
Browse files Browse the repository at this point in the history
…mentation into fix_vit_pos_embed
  • Loading branch information
xiexinch committed Jul 19, 2021
2 parents 523c440 + bab4c03 commit 9a4e1e8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mmseg/models/backbones/vit.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,8 +118,8 @@ class VisionTransformer(BaseModule):
attn_drop_rate (float): The drop out rate for attention layer.
Default 0.0
drop_path_rate (float): stochastic depth rate. Default 0.0
with_cls_token (bool): If concatenating class token into image tokens
as transformer input. Default: True.
with_cls_token (bool): Whether concatenating class token into image
tokens as transformer input. Default: True.
output_cls_token (bool): Whether output the cls_token. If set True,
`with_cls_token` must be True. Default: False.
norm_cfg (dict): Config dict for normalization layer.
Expand Down

0 comments on commit 9a4e1e8

Please sign in to comment.