fix gpu
Browse files- hyperparams.yaml +1 -2
hyperparams.yaml
CHANGED
|
@@ -34,8 +34,6 @@ eos_index: 2
|
|
| 34 |
# Decoding parameters
|
| 35 |
min_decode_ratio: 0.0
|
| 36 |
max_decode_ratio: 1.0
|
| 37 |
-
valid_search_interval: 10
|
| 38 |
-
valid_beam_size: 1 # We do greedy here so it's faster to decode ...
|
| 39 |
test_beam_size: 80
|
| 40 |
ctc_weight_decode: 0.3
|
| 41 |
scorer_beam_scale: 0.3
|
|
@@ -124,6 +122,7 @@ modules:
|
|
| 124 |
encoder: !ref <encoder>
|
| 125 |
transformer: !ref <Transformer>
|
| 126 |
decoder: !ref <decoder>
|
|
|
|
| 127 |
|
| 128 |
log_softmax: !new:torch.nn.LogSoftmax
|
| 129 |
dim: -1
|
|
|
|
| 34 |
# Decoding parameters
|
| 35 |
min_decode_ratio: 0.0
|
| 36 |
max_decode_ratio: 1.0
|
|
|
|
|
|
|
| 37 |
test_beam_size: 80
|
| 38 |
ctc_weight_decode: 0.3
|
| 39 |
scorer_beam_scale: 0.3
|
|
|
|
| 122 |
encoder: !ref <encoder>
|
| 123 |
transformer: !ref <Transformer>
|
| 124 |
decoder: !ref <decoder>
|
| 125 |
+
ctc_lin: !ref <ctc_lin>
|
| 126 |
|
| 127 |
log_softmax: !new:torch.nn.LogSoftmax
|
| 128 |
dim: -1
|