5 Commits (26634db7a8d9c5c1faa1c2b06f3a34006e835d46)

Author SHA1 Message Date
  Megvii Engine Team f5cb21ed3a fix(mgb/opr): add non finite check 4 years ago
  Megvii Engine Team c5f8b58385 docs(docstring): transfer to google style 4 years ago
  Megvii Engine Team bfc4e7a966 docs(mge): fix amp docstring problems 4 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 4 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 4 years ago