Ma Zerun 556fa567a8

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

* [Feature] Support MViT.

* Add MViT configs and docs

* Add unit test

* Fix unit tests.

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

..

data

…

test_data

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

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

test_downstream

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

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

test_metrics

[Enhance] Use PyTorch official one_hot to implement convert_to_one_hot. (#696)

2022-02-28 10:20:31 +08:00

test_models

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

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

test_runtime

…

test_utils

[Feature] Support MPS device. (#894)

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