This repository has been archived by the owner on Jul 22, 2024. It is now read-only.

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 subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet