sennnnn 5c195db1bd 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

..

__init__.py

support transformer backbone (#465)

2021-04-21 20:19:55 -07:00

cgnet.py

Fix typo: upsampe_cfg -> upsample_cfg (#449)

2021-03-30 09:51:38 -07:00

fast_scnn.py

[Enhance] Refactor inverted residual (#164)

2020-09-28 00:33:51 +08:00

hrnet.py

Fix typo: upsampe_cfg -> upsample_cfg (#449)

2021-03-30 09:51:38 -07:00

mobilenet_v2.py

[Enhance] Refactor inverted residual (#164)

2020-09-28 00:33:51 +08:00

mobilenet_v3.py

Fix typo: upsampe_cfg -> upsample_cfg (#449)

2021-03-30 09:51:38 -07:00

resnest.py

Support ResNeSt backbone (#47)

2020-08-17 00:54:01 +08:00

resnet.py

Add Colab Tutorial (#7)

2020-07-10 16:55:47 +08:00

resnext.py

init commit

2020-07-10 02:39:01 +08:00

unet.py

Save base_channels for UNet (#450)

2021-03-30 09:52:08 -07:00

vit.py

Add option for output shape of ViT (#530)

2021-05-05 22:49:28 -07:00