From 4bd348b2ab4eb9458c158cfbae3424239d9ccf7b Mon Sep 17 00:00:00 2001 From: zhouneng Date: Sat, 8 May 2021 12:00:34 +0800 Subject: [PATCH] modify base learning rate to improve accuracy --- model_zoo/official/nlp/textcnn/sst2_config.yaml | 2 +- model_zoo/official/nlp/textcnn/subj_config.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/nlp/textcnn/sst2_config.yaml b/model_zoo/official/nlp/textcnn/sst2_config.yaml index 8231052ea39..a5d3f72bb70 100644 --- a/model_zoo/official/nlp/textcnn/sst2_config.yaml +++ b/model_zoo/official/nlp/textcnn/sst2_config.yaml @@ -24,7 +24,7 @@ checkpoint_path: './checkpoint/' checkpoint_file_path: 'train_textcnn-4_149.ckpt' word_len: 51 vec_length: 40 -base_lr: 1e-3 +base_lr: 5e-3 # Export options device_id: 0 diff --git a/model_zoo/official/nlp/textcnn/subj_config.yaml b/model_zoo/official/nlp/textcnn/subj_config.yaml index d65fd72296e..0a81f9729b8 100644 --- a/model_zoo/official/nlp/textcnn/subj_config.yaml +++ b/model_zoo/official/nlp/textcnn/subj_config.yaml @@ -24,7 +24,7 @@ checkpoint_path: './checkpoint/' checkpoint_file_path: 'train_textcnn-4_149.ckpt' word_len: 51 vec_length: 40 -base_lr: 1e-3 +base_lr: 8e-4 # Export options device_id: 0