update config

2019-12-05 23:50:10 +08:00 · 2019-12-05 23:50:10 +08:00 · fabcef1825
parent 276e3d5569
commit fabcef1825
2 changed files with 2 additions and 2 deletions
--- a/conf/config.yaml
+++ b/conf/config.yaml
@ -11,6 +11,6 @@ defaults:
  - preprocess
  - train
  - embedding
-  - model: capsule
+  - model: cnn   # [cnn, rnn, transformer, capsule, gcn, lm]


--- a/conf/train.yaml
+++ b/conf/train.yaml
@ -3,7 +3,7 @@ seed: 1
 use_gpu: True
 gpu_id: 0

-epoch: 5
+epoch: 50
 batch_size: 32
 learning_rate: 3e-4
 lr_factor: 0.7 # 学习率的衰减率