onnxruntime
1ea462fc
- put sce and scr grad into fp32
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 years ago
put sce and scr grad into fp32 put sce as fp32 node instead of loss node sce gradient doesn't have gpu-fp16 implementation, so also insert cast op for it
Author
zhijxu-MS
Parents
39eb6111
Loading