#知りたいこと
2つの別々のモデルを同時に学習させる方法
やりたいこと
2つのモデルを用意し,学習の過程で各モデルがお互いの中間層出力の値を損失関数にいれ学習させたいです.
つまり,2つのモデルで相互作用しながら学習させたいということです.
#試したこと
下記のように2つのモデルA,B,自作の損失関数式を定義し,別々に学習させるということをしました.
ただ,このコードだと,モデルA→モデルBという順番で学習されてしまいます.
私は,順番ではなく同時に学習させていきたいと考えています.(損失関数は別々の関数を使いたいです)
python
1A_model.compile(optimizer='adam', loss=mylossfunc_A) 2B_model.compile(optimizer='adam', loss=mylossfunc_B) 3 4A_model.fit(a_train, a_test, 5 epochs=n_epoch, 6 batch_size=batch_size, 7 verbose=verbose, 8 shuffle=True, 9 validation_split=0.1, 10 ) 11 12B_model.fit(b_train, b_test, 13 epochs=n_epoch, 14 batch_size=batch_size, 15 verbose=verbose, 16 shuffle=True, 17 validation_split=0.1, 18 ) 19
あなたの回答
tips
プレビュー