Skip to content

Commit

Permalink
README update
Browse files Browse the repository at this point in the history
  • Loading branch information
Bernhard Walser committed Mar 29, 2021
1 parent 4d5cb7c commit 47e7e56
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ What dimension each head should have.
What window size to use, make sure that after each downscaling the image dimensions are still divisible by the window size.
- `downscaling_factors`: 4-tuple of ints.
What downscaling factor to use in each stage. Make sure image dimension is large enough for downscaling factors.
- `relative_position_embedding`: bool.
- `relative_pos_embedding`: bool.
Whether to use learnable relative position embedding (2M-1)x(2M-1) or full positional embeddings (M²xM²).
### TODO
- Adjust code for and validate on ImageNet-1K and COCO 2017
Expand Down

0 comments on commit 47e7e56

Please sign in to comment.