混合精度の利用 + bfloat16のオプション追加

[nadare881]

RVCの訓練では混合精度を利用しています。
混合精度はモデルのパラメータをfloat32で保存し、計算の数値だけfloat16で計算することでfloat32の精度とfloat16の速度のいいとこどりをする手法です。
https://www.tensorflow.org/guide/mixed_precision?hl=ja
現状学習のコードを見ると、float16利用の時モデルの方もfloat16にしてしまっているので修正します。

• 指数部がfloat32で仮数部を小さくすることでfloat32と同じ桁の幅を表現できるbfloat16というのもあり、transformerではこちらの方が安定しやすいのでoptionにbfloat16も追加し選べるようにします。

緊急ではないのでゆっくりめに進めます。