sennnnn aa9b609f11
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;
2021-05-05 22:49:28 -07:00
..
2020-07-10 02:39:01 +08:00
2021-04-30 11:34:57 -07:00
2021-03-30 17:55:09 -07:00
2021-05-05 16:56:19 -07:00
2021-04-30 11:34:57 -07:00
2020-08-09 23:49:23 +08:00