損失関数の呼ばれた回数をカウントしたいです

Question

##質問
kerasで損失関数を自作しました．その中で，損失関数が呼ばれた回数を使いたいと思いまして，以下のようにcountに逐次+1をするようにコードを書きました．しかし，最後のprint(count)の出力が1になってしまいます．
損失関数は1回しか呼ばれてないのでしょうか.

##code

count = 0

def encoder(input_):
    d1 = Dense(3, activation='relu', name='encoder_input')(input_)
    d2 = Dense(2, 'encoder_output')(d1)
    return d2

def decoder(input_):
    d1 = Dense(3, activation='relu', name='decoder_input')(input_)
    d2 = Dense(2, name='decoder_output')(d1)
    return d2

def my_loss_function(y_pred, y_true):
    global count
    count += 1
    return K.mean(K.square(y_pred - y_true), axis=-1)  

# input
input = Input(shape=(2,))

# output
output = decoder(encoder(input))

# model
model = Model(input=input, output=output)

model.compile(optimizer='adam', loss='my_loss_function')
model_hist = model.fit(x_train, x_test,
                        epochs=n_epoch,
                        batch_size=1,
                        verbose=verbose,
                        shuffle=True)

print('count=',count)
#count=1

Answer 1

参考記事のように他にも関数の呼び出し関数をカウントする方法はあるので、それらを試してみて同じ結果(カウント数1)になれば、「損失関数は1回しか呼ばれてない」と言えるかと思います。

参考記事：【python】呼び出し回数カウント関数を色々な方法で作る

Comments

[cdg_26]

回答ありがとうございます．
いくつかの方法でカウントしてみた結果，損失関数は1回しか呼ばれていないようでした．（バッチサイズを1に設定していたので，訓練データの数だけ呼ばれていると予想していましたが）

関連して，追加で質問させていただきたいのですが，
今回実現したいことが，1ステップ（1つのデータを読み込み損失を計算する）ごと，外部で用意した配列の要素を指定し損失関数に組み込むということです．
ですので，1stepごと配列の要素を指定するindexの値が必要だと考えています．
上のカスタム損失関数の中でindexを得る方法，もしくは，それを実現できる方法をご存知でしたら教えていただきたいです．

[amahara_waya]

それに関してはすでに同じ質問を立てているみたいなので、別件としてそちらで消化してもらえると幸いです。
https://teratail.com/questions/203376

[cdg_26]

承知しました
また，なにかご存知なことがありましたら，ぜひそちらにコメントお待ちしております．