KL-lossの前につける係数として、経過した学習のステップ数をKerasで持ってくる方法をご存知な方がいれば教えていただけると助かります。