❌ This version of handling ViT activation (through trucation) is problematic! The latest version will use the pytorch built-in method for saving and tracking activation and gradient.#2
Open
TyBruceChen wants to merge 10 commits into
Enhance your code review process with GitHub Actions
GitHub Actions make it easy to automate all your software workflows, now with world-class CI/CD.
Build, test, and deploy your code right from GitHub. Learn more about GitHub Actions.
Linux, macOS, Windows, and containers
Matrix builds
Any language
Live logs
Built-in secret store
