Hakjin Lee 05e4bc17b2

[Feature] Support Activation Checkpointing for ConvNeXt. (#1152) ...

* Support Activation Checkpointing for ConvNeXt

* Add test case

* Lint

* Add docstring

2022-11-14 17:08:31 +08:00

..

__init__.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_conformer.py

…

test_convmixer.py

[Feature] Support ConvMixer. (#716)

2022-03-23 15:23:57 +08:00

test_convnext.py

[Feature] Support Activation Checkpointing for ConvNeXt. (#1152)

2022-11-14 17:08:31 +08:00

test_cspnet.py

[Feature] Add CSPNet and backbone and checkpoints (#735)

2022-03-30 15:29:35 +08:00

test_deit.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_densenet.py

[Feature] Support DenseNet. (#750)

2022-04-28 13:06:03 +08:00

test_efficientformer.py

[Feature] Support HorNet Backbone. (#1013)

2022-09-27 10:37:49 +08:00

test_efficientnet.py

…

test_hornet.py

[Feature] Support HorNet Backbone. (#1013)

2022-09-27 10:37:49 +08:00

test_hrnet.py

…

test_mlp_mixer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_mobilenet_v2.py

…

test_mobilenet_v3.py

…

test_mvit.py

[Feature] Support MViT and add checkpoints. (#924)

2022-08-02 15:20:16 +08:00

test_poolformer.py

[Feature] Add PoolFormer backbone and checkpoints. (#746)

2022-04-15 11:35:15 +08:00

test_regnet.py

…

test_repmlp.py

[Feature] ADD RepMLP Backbone and checkpoints. (#709)

2022-03-30 20:40:25 +08:00

test_repvgg.py

[Enhance] RepVGG for YOLOX-PAI. (#1025)

2022-09-30 14:20:53 +08:00

test_res2net.py

…

test_resnest.py

…

test_resnet_cifar.py

…

test_resnet.py

…

test_resnext.py

…

test_seresnet.py

…

test_seresnext.py

…

test_shufflenet_v1.py

…

test_shufflenet_v2.py

…

test_swin_transformer_v2.py

[Feature] Support Swin Transform V2. (#799)

2022-08-03 17:33:08 +08:00

test_swin_transformer.py

[Fix] Fix the output position of Swin-Transformer. (#947)

2022-08-03 19:32:29 +08:00

test_t2t_vit.py

[Refactor] Re-write get_sinusoid_encoding from third-party implementation. (#965)

2022-09-13 15:24:29 +08:00

test_timm_backbone.py

[Feature] Support MPS device. (#894)

2022-07-28 12:28:51 +08:00

test_tnt.py

…

test_twins.py

…

test_van.py

[Feature] Support VAN. (#739)

2022-04-28 09:35:17 +08:00

test_vgg.py

…

test_vision_transformer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

utils.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00