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'