From fabcef1825f385d0f361387c1b16df3246867928 Mon Sep 17 00:00:00 2001
From: leo <yuwl798180@163.com>
Date: Thu, 5 Dec 2019 23:50:10 +0800
Subject: [PATCH] update config

---
 conf/config.yaml | 2 +-
 conf/train.yaml  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/conf/config.yaml b/conf/config.yaml
index bf6dce9..0d82724 100644
--- 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]
 
 
diff --git a/conf/train.yaml b/conf/train.yaml
index d7b25f4..7bf4624 100644
--- 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 # 学习率的衰减率