search

BR-IDL / PaddleViT

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+
https://github.com/BR-IDL/PaddleViT
Apache License 2.0
1.22k stars 318 forks source link

some models can not enable amp mode #101

Closed SigureMo closed 2 years ago

SigureMo commented 2 years ago

https://github.com/BR-IDL/PaddleViT/blob/796043557bb617abb22fa546876000cc0c8946e4/image_classification/BEiT/config.py#L170-L174

部分模型（如这里的 BEiT）貌似遗漏了开启 amp 相关逻辑代码（else 里）

cjh3020889729 commented 2 years ago

Thanks for your Pull Request!

After we check the problem, we will merge your PR.