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

Honoring distributed flag + fixing reset_classifier #6

Open

abhrac wants to merge 1 commit into IBM:main from abhrac:main

abhrac commented Jan 12, 2022

Honoring the args.distributed flag in calls to evaluate().
A couple of changes to make the reset_classifier() method work:

Initializing the embed_dim instance variable in VisionTransformer.
Reinitializing the classification head for all branches.


          honoring distributed flag + fixing reset_classifier

97a1414

Author

abhrac commented Jan 17, 2022

@chunfuchen could you please have a look?

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