New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fixing potential calculation error mentioned in #28 and accelerating shuffling process #32

Open

huixiancheng wants to merge 3 commits into chaytonmin:main from huixiancheng:main

huixiancheng commented Dec 19, 2023

Fixing potential calculation error mentioned in Understanding Range Aware Masking? #28.
Speeds up the shuffling process, in my tests it takes about 50% time of a forward. Modified training speed up by 50%.
However, Please test it before the update.

huixiancheng added 3 commits

December 19, 2023 05:04


          Update voxel_mae.py

adecbb3


          Update voxel_mae.py

b81faea


          Update voxel_mae_res.py

e617b1b

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet