blob: 002e2b2bca728e61a46d08f41cf5727178bb77a5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
from typing import List
from typing import NamedTuple
from typing import Optional
class CBHGDiscriminatorModelConfig(NamedTuple):
in_channels: int
hidden_channels_list: List[int]
class CBHGModelConfig(NamedTuple):
in_channels: int
conv_bank_out_channels: int
conv_bank_k: int
max_pooling_k: int
conv_projections_hidden_channels: int
highway_layers: int
out_channels: int
out_size: int
aligner_out_time_length: int
disable_last_rnn: bool
enable_aligner: bool
discriminator: Optional[CBHGDiscriminatorModelConfig]
class CBHGLossConfig(NamedTuple):
l1: float
predictor_fake: float
discriminator_true: float
discriminator_fake: float
discriminator_grad: float
|
