summaryrefslogtreecommitdiff
path: root/become_yukarin/config.py
diff options
context:
space:
mode:
authorHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2017-11-20 03:06:39 +0900
committerHiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>2017-11-20 03:06:39 +0900
commit16b4e72fe6728e2e64d4c6357b7c73ac06868c1c (patch)
tree657f0398b9a237ab46327d08f58a230b9581669b /become_yukarin/config.py
parent437a869590c989c184d33990b1d788149d073ee9 (diff)
aligner
Diffstat (limited to 'become_yukarin/config.py')
-rw-r--r--become_yukarin/config.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/become_yukarin/config.py b/become_yukarin/config.py
index 05b0790..07f35fd 100644
--- a/become_yukarin/config.py
+++ b/become_yukarin/config.py
@@ -32,6 +32,8 @@ class ModelConfig(NamedTuple):
highway_layers: int
out_channels: int
out_size: int
+ aligner_out_time_length: int
+ disable_last_rnn: bool
class LossConfig(NamedTuple):
@@ -100,6 +102,8 @@ def create_from_json(s: Union[str, Path]):
highway_layers=d['model']['highway_layers'],
out_channels=d['model']['out_channels'],
out_size=d['model']['out_size'],
+ aligner_out_time_length=d['model']['aligner_out_time_length'],
+ disable_last_rnn=d['model']['disable_last_rnn'],
),
loss=LossConfig(
l1=d['loss']['l1'],