Ezra-Yu e54cfd6951

[Imporve] Using train_step instead of forward in PreciseBNHook (#964) ...

* fix precise BN hook when using MLU

* fix unit tests

2022-08-11 15:02:25 +08:00

..

data

[CI] Add Windows CI. (#708)

2022-02-28 10:06:08 +08:00

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 CSRA head. (#881)

2022-08-04 18:15:51 +08:00

test_runtime

[Imporve] Using train_step instead of forward in PreciseBNHook (#964)

2022-08-11 15:02:25 +08:00

test_utils

[Feature] Support MPS device. (#894)

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