tests/test_cnn/test_generalized_attention.py · 5be95934990f7056662ef0109066f0d35d03ab3e · OpenDAS / MMCV

Guangchen Lin authored May 23, 2021

* fix generalized attention fp16

* fix building without gpu error

* add comment

* Cast tensor at initialization

5be95934

test_generalized_attention.py 2.78 KB