search

FrankWork / fudan_mtl_reviews

TensorFlow implementation of the paper `Adversarial Multi-task Learning for Text Classification`
178 stars 40 forks source link

添加summaries，每次训练也restore #13

Closed JaysonAlbert closed 5 years ago

JaysonAlbert commented 5 years ago

  1. 使用summary可以方便的在tensorboard里面看各种loss的情况，方便调超参数。

  2. 原本代码中save的时候没有保存所有的变量，因为set_saver放到了build_train_op前面。

JaysonAlbert commented 5 years ago

在我的那个代码里还添加了个lstm版本的实现，以及增加了attention，由于改动较大，就没有pull request，有需要的话可以@我

JaysonAlbert commented 5 years ago

其中mean目录下是所有数据集平均的结果