moiduy04 commited on
Commit
75c31f0
1 Parent(s): 5d0c765

Update config/config_final.yaml

Browse files
Files changed (1) hide show
  1. config/config_final.yaml +2 -2
config/config_final.yaml CHANGED
@@ -20,8 +20,8 @@ model:
20
  num_encoder_layers: 8
21
  num_decoder_layers: 4
22
  model_folder: "weights"
23
- model_basename: "transformer_"
24
- preload: 'huge_61'
25
 
26
  train:
27
  lr: 0.001 # 1e-2
 
20
  num_encoder_layers: 8
21
  num_decoder_layers: 4
22
  model_folder: "weights"
23
+ model_basename: "transformer_huge"
24
+ preload: '_61'
25
 
26
  train:
27
  lr: 0.001 # 1e-2