Update train.yaml
This commit is contained in:
parent
4024d91226
commit
c382a7806c
|
@ -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'
|
||||
load_path: './model_bert.pt'
|
||||
|
|
Loading…
Reference in New Issue