Skip to content
Snippets Groups Projects
Commit 5c195db1 authored by sennnnn's avatar sennnnn Committed by GitHub
Browse files

Add option for output shape of ViT (#530)

* Add arg: final_reshape to control if converting output feature information from NLC to NCHW;

* Fix the default value of final_reshape;

* Modify arg: final_reshape to arg: out_shape;

* Fix some unit test bug;
parent 0d477ac5
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment