mmpretrain/.circleci
Ma Zerun b017670e1b
[Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`. (#1434)
* [Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`.

* Support `--local-rank` and `--amp` option for new version PyTorch.

* Fix imports and UT.
2023-03-29 15:50:44 +08:00
..
docker [CI] Add Circle CI for mmcls 1.x. (#970) 2022-08-22 15:02:08 +08:00
config.yml Fix CI 2023-02-17 15:33:16 +08:00
test.yml [Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`. (#1434) 2023-03-29 15:50:44 +08:00