What is the difference between trans_dim and encoder_dims?

Pang-Yatian / Point-MAE

[ECCV2022] Masked Autoencoders for Point Cloud Self-supervised Learning

MIT License

460 stars 55 forks source link

What is the difference between trans_dim and encoder_dims? #15

Closed Phoebe-ovo closed 2 years ago

Pang-Yatian commented 2 years ago

Hi,

trans_dim refers to Transformer dimension. encoder_dims refers to the output dimension of the token embedding module. In our code, they are set to the same value.

Hope this clarifies.