From 1f8c059860a7f18192c62414e92c38d824a1b561 Mon Sep 17 00:00:00 2001
From: tlk-dsg <467460833@qq.com>
Date: Wed, 29 Sep 2021 16:35:34 +0800
Subject: [PATCH] test

---
 example/ner/few-shot/conf/config.yaml  |  3 +-
 example/ner/few-shot/conf/predict.yaml | 45 +++++++++++++-------------
 2 files changed, 24 insertions(+), 24 deletions(-)

diff --git a/example/ner/few-shot/conf/config.yaml b/example/ner/few-shot/conf/config.yaml
index c14246f..0b99422 100644
--- a/example/ner/few-shot/conf/config.yaml
+++ b/example/ner/few-shot/conf/config.yaml
@@ -1,4 +1,5 @@
 cwd: ???
 
 defaults:
-  - train/conll
\ No newline at end of file
+  - train/conll
+  - predict
\ No newline at end of file
diff --git a/example/ner/few-shot/conf/predict.yaml b/example/ner/few-shot/conf/predict.yaml
index c5edb00..2ffbec9 100644
--- a/example/ner/few-shot/conf/predict.yaml
+++ b/example/ner/few-shot/conf/predict.yaml
@@ -1,29 +1,28 @@
 cwd: ???
 
-defaults:
-  seed: 1
+seed: 1
 
-  bart_name: "facebook/bart-large"
-  dataset_name: conll2003
-  device: cuda
+bart_name: "facebook/bart-large"
+dataset_name: conll2003
+device: cuda
 
-  num_epochs: 30
-  batch_size: 16
-  learning_rate: 2e-5
-  warmup_ratio: 0.01
-  eval_begin_epoch: 16
-  src_seq_ratio: 0.6
-  tgt_max_len: 10
-  num_beams: 1
-  length_penalty: 1
+num_epochs: 30
+batch_size: 16
+learning_rate: 2e-5
+warmup_ratio: 0.01
+eval_begin_epoch: 16
+src_seq_ratio: 0.6
+tgt_max_len: 10
+num_beams: 1
+length_penalty: 1
 
-  use_prompt: True
-  prompt_len: 10
-  prompt_dim: 800
+use_prompt: True
+prompt_len: 10
+prompt_dim: 800
 
-  freeze_plm: True
-  learn_weights: True
-  notes: ''
-  save_path: null # 模型保存路径
-  load_path: load_path # 模型加载路径，不能为空
-  write_path: "data/conll2003/predict.txt"
\ No newline at end of file
+freeze_plm: True
+learn_weights: True
+notes: ''
+save_path: null # 模型保存路径
+load_path: load_path # 模型加载路径，不能为空
+write_path: "data/conll2003/predict.txt"
\ No newline at end of file