From c382a7806c3b0524fce150bd8fb564cbb74442a7 Mon Sep 17 00:00:00 2001
From: TimelordRi <35120358+TimelordRi@users.noreply.github.com>
Date: Sun, 10 Oct 2021 20:46:10 +0800
Subject: [PATCH] Update train.yaml

---
 example/re/few-shot/conf/train.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/example/re/few-shot/conf/train.yaml b/example/re/few-shot/conf/train.yaml
index 6490111..68e7557 100644
--- a/example/re/few-shot/conf/train.yaml
+++ b/example/re/few-shot/conf/train.yaml
@@ -41,7 +41,7 @@ max_time: None
 min_epochs: None
 min_steps: None
 model_class: 'BertForMaskedLM'
-model_name_or_path: 'bert-large-uncased'
+model_name_or_path: 'bert-base-uncased'
 move_metrics_to_cpu: False
 multiple_trainloader_mode: 'max_size_cycle'
 num_nodes: 1
@@ -80,4 +80,4 @@ wandb: False
 weight_decay: 0.01
 weights_save_path: None
 weights_summary: 'top'
-load_path: './model_bert.pt'
\ No newline at end of file
+load_path: './model_bert.pt'