mmengine/runner/amp.py · 96378fa748cea41d5c1a6ce933643411cab2d4da · Florian Schiffel / ICV-mmengine_basecode

2 years ago

[Fix] make `autocast` compatible with GTX1660 and make it more robust. (#344) · 96378fa7

Mashiro authored 2 years ago

* fix amp

* fix amp

* make logic consistent with torch.autocast

* support multiple device

* fix as comment

* fix as comment

* avoid circle import

96378fa7

History

[Fix] make `autocast` compatible with GTX1660 and make it more robust. (#344)

Mashiro authored 2 years ago

* fix amp

* fix amp

* make logic consistent with torch.autocast

* support multiple device

* fix as comment

* fix as comment

* avoid circle import