Update configuration_translation_transformer.py
Browse files
configuration_translation_transformer.py
CHANGED
|
@@ -6,7 +6,7 @@ class TranslationTransformerConfig(PretrainedConfig):
|
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
vocab_size=32000,
|
| 9 |
-
d_model=
|
| 10 |
nhead=8,
|
| 11 |
num_encoder_layers=3,
|
| 12 |
num_decoder_layers=3,
|
|
|
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
vocab_size=32000,
|
| 9 |
+
d_model=256,
|
| 10 |
nhead=8,
|
| 11 |
num_encoder_layers=3,
|
| 12 |
num_decoder_layers=3,
|