JinSakuma / TGNN

1 stars 0 forks source link

SWT の追加 #1

Open HayatoKTYM opened 3 years ago

HayatoKTYM commented 3 years ago

目的

学習時に CTR(alpha) だけでなく SWT(u) を更新できるように拡張

タスク

swt 部分は更新するかしないかを指定できるようにする (argparseなど)

注意点

SWT 学習時の 教師ラベルは uA, uB で別々に与える必要あり.

HayatoKTYM commented 3 years ago

<問題> swt 出力を使うようにすると,forward関数になってからu_predが得られるので,ここが適用できない https://github.com/JinSakuma/TGNN/blob/master/utils/utils_ctc.py#L113

HayatoKTYM commented 3 years ago

<対応策> TO Do

HayatoKTYM commented 3 years ago

SWT に読み込む重みのPATH

/mnt/aoni04/katayama/share/SPEC/epoch_20_acc0.887_loss0.266_ut_train.pth

HayatoKTYM commented 3 years ago

上記の話は SWT 更新を考えるまで一旦保留