11 Commits (fc0f454685ec94ff46e591376629b7518a58efb6)

Author SHA1 Message Date
  Megvii Engine Team fc0f454685 fix(dnn/check_non_finite): adjust some details of CheckNonFinite 3 years ago
  Megvii Engine Team e393d1cf65 feat(mge/amp): add convert_format module for NHWC training 4 years ago
  Megvii Engine Team c2e9860feb chore(license): remove all license in file header 3 years ago
  Megvii Engine Team 3c3fc6f33c refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++ 3 years ago
  Megvii Engine Team 2881934cb8 feat(dnn/check_non_finite): addmul scale to check_non_finite opr 4 years ago
  Megvii Engine Team 11d75fecb5 feat(dnn/check_non_finite): add batch check_non_finite 4 years ago
  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