Update config/config_huge.yaml
Browse files- 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: "
|
24 |
-
preload: '
|
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
|