今研究しているコードの損失関数の埋め込み方

＜募集例＞

【概要】
今機械学習で研究を行っておりこのコードの損失関数の入れ方について教えていただければと思います。

【目的】
損失関数を既存のものに追加すること目指しておりそこでつまづいています。

【募集背景】
現在は時間がないという状況で、損失関数を改善するにあたり、助けが必要となりました。

【コミュニケーション方法】
音声通話とチャットの両方（音声のほうが伝わりやすいと思うので音声）希望

【参考URL】
https://github.com/Vandermode/ERRNet(もとになるコード）
https://github.com/ceciliavision/perceptual-reflection-removal（組み込みたい損失関数exclutionlossがあるコード）

【ご提案時のお願い】
提案いただく際は変えた部分についてご提示頂けますと検討がしやすいです。宜しくお願いいたします。