SkyID
/
deepke
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update train.yaml

This commit is contained in:
TimelordRi 2021-10-10 20:46:10 +08:00 committed by GitHub
parent 4024d91226
commit c382a7806c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example/re/few-shot/conf/train.yaml
View File

@ -41,7 +41,7 @@ max_time: None
min_epochs: None min_epochs: None
min_steps: None min_steps: None
model_class: 'BertForMaskedLM' model_class: 'BertForMaskedLM'
model_name_or_path: 'bert-large-uncased' model_name_or_path: 'bert-base-uncased'
move_metrics_to_cpu: False move_metrics_to_cpu: False
multiple_trainloader_mode: 'max_size_cycle' multiple_trainloader_mode: 'max_size_cycle'
num_nodes: 1 num_nodes: 1
@ -80,4 +80,4 @@ wandb: False
weight_decay: 0.01 weight_decay: 0.01
weights_save_path: None weights_save_path: None
weights_summary: 'top' weights_summary: 'top'
load_path: './model_bert.pt' load_path: './model_bert.pt'