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