Training Neural Networks with BFloat16

Thank you for your answer.
I’ve tried using Automatic Mixed Precision, and it works.

Best Regards,
Rodrigo