Open HayatoKTYM opened 3 years ago
<問題> swt 出力を使うようにすると,forward関数になってからu_predが得られるので,ここが適用できない https://github.com/JinSakuma/TGNN/blob/master/utils/utils_ctc.py#L113
<対応策> TO Do
/mnt/aoni04/katayama/share/SPEC/epoch_20_acc0.887_loss0.266_ut_train.pth
上記の話は SWT 更新を考えるまで一旦保留
目的
学習時に CTR(alpha) だけでなく SWT(u) を更新できるように拡張
タスク
swt 部分は更新するかしないかを指定できるようにする (argparseなど)
注意点
SWT 学習時の 教師ラベルは uA, uB で別々に与える必要あり.