3 Commits (c25125e3d2c0d69c283dfb502d08895601110c45)

Author SHA1 Message Date
  Megvii Engine Team bfc4e7a966 docs(mge): fix amp docstring problems 4 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 5 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 5 years ago