Skip to content
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

Add option for output shape of ViT #530

Merged
merged 4 commits into from
May 6, 2021
Merged

Add option for output shape of ViT #530

merged 4 commits into from
May 6, 2021

Commits on May 6, 2021

  1. Add arg: final_reshape to control if converting output feature inform… 

    …ation from NLC to NCHW;
    sennnnn committed May 6, 2021
    Configuration menu
    Copy the full SHA
    be0d2fb View commit details
    Browse the repository at this point in the history

  2. Fix the default value of final_reshape;

    sennnnn committed May 6, 2021
    Configuration menu
    Copy the full SHA
    49130ce View commit details
    Browse the repository at this point in the history

  3. Modify arg: final_reshape to arg: out_shape;

    sennnnn committed May 6, 2021
    Configuration menu
    Copy the full SHA
    f7052f9 View commit details
    Browse the repository at this point in the history

  4. Fix some unit test bug;

    sennnnn committed May 6, 2021
    Configuration menu
    Copy the full SHA
    c5858f5 View commit details
    Browse the repository at this point in the history