Browse Source

modify base learning rate to improve accuracy

pull/16094/head
zhouneng 4 years ago
parent
commit
4bd348b2ab
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'
word_len: 51
vec_length: 40
base_lr: 1e-3
base_lr: 5e-3

# Export options
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'
word_len: 51
vec_length: 40
base_lr: 1e-3
base_lr: 8e-4

# Export options
device_id: 0


Loading…
Cancel
Save