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

..

data

…

test_data

[Feature] Support Stanford Cars dataset. (#893)

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

test_downstream

…

test_metrics

…

test_models

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

2022-11-14 17:08:31 +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