moiduy04 commited on
Commit
6c5a1e1
1 Parent(s): 75c31f0

Update config/config_huge.yaml

Browse files
Files changed (1) hide show
  1. config/config_huge.yaml +2 -2
config/config_huge.yaml CHANGED
@@ -20,8 +20,8 @@ model: # 61604879 parameters
20
  num_encoder_layers: 8
21
  num_decoder_layers: 4
22
  model_folder: "weights"
23
- model_basename: "transformer_"
24
- preload: 'huge_50'
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: '_50'
25
 
26
  train:
27
  lr: 0.001 # 1e-2