Browse Source

!16094 modify learning rate for textcnn with datasets sst2 and subj

From: @zhouneng2
Reviewed-by: @c_34,@liangchenghui
Signed-off-by: @c_34
pull/16094/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
6f36f4a43b
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      model_zoo/official/nlp/textcnn/sst2_config.yaml
  2. +1
    -1
      model_zoo/official/nlp/textcnn/subj_config.yaml

+ 1
- 1
model_zoo/official/nlp/textcnn/sst2_config.yaml View File

@@ -24,7 +24,7 @@ checkpoint_path: './checkpoint/'
checkpoint_file_path: 'train_textcnn-4_149.ckpt' checkpoint_file_path: 'train_textcnn-4_149.ckpt'
word_len: 51 word_len: 51
vec_length: 40 vec_length: 40
base_lr: 1e-3
base_lr: 5e-3


# Export options # Export options
device_id: 0 device_id: 0


+ 1
- 1
model_zoo/official/nlp/textcnn/subj_config.yaml View File

@@ -24,7 +24,7 @@ checkpoint_path: './checkpoint/'
checkpoint_file_path: 'train_textcnn-4_149.ckpt' checkpoint_file_path: 'train_textcnn-4_149.ckpt'
word_len: 51 word_len: 51
vec_length: 40 vec_length: 40
base_lr: 1e-3
base_lr: 8e-4


# Export options # Export options
device_id: 0 device_id: 0


Loading…
Cancel
Save