You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
322 B
22 lines
322 B
3 years ago
|
model_module: transformer
|
||
3 years ago
|
|
||
|
|
data:
|
||
|
|
unit_type: spm
|
||
|
|
|
||
|
3 years ago
|
model:
|
||
|
|
n_vocab: 5002
|
||
|
|
pos_enc: null
|
||
|
|
embed_unit: 128
|
||
|
|
att_unit: 512
|
||
|
|
head: 8
|
||
|
|
unit: 2048
|
||
|
|
layer: 16
|
||
|
|
dropout_rate: 0.5
|
||
|
|
emb_dropout_rate: 0.0
|
||
|
|
att_dropout_rate: 0.0
|
||
|
|
tie_weights: False
|
||
|
3 years ago
|
|
||
|
|
decoding:
|
||
|
|
batch_size: 30
|
||
|
|
num_workers: 2
|