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

[CI] Add Windows CI. (#708)

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

test_data

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

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