File size: 600 Bytes
32b2aaa 1df74c6 32b2aaa 1df74c6 32b2aaa 1df74c6 |
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 |
from dataclasses import dataclass
from pathlib import Path
from typing import Union
from ..hparams import HParams as HParamsBase
@dataclass(frozen=True)
class HParams(HParamsBase):
cfm_solver_method: str = "midpoint"
cfm_solver_nfe: int = 64
cfm_time_mapping_divisor: int = 4
univnet_nc: int = 96
lcfm_latent_dim: int = 64
lcfm_training_mode: str = "ae"
lcfm_z_scale: float = 5
vocoder_extra_dim: int = 32
gan_training_start_step: Union[int, None] = 5_000
enhancer_stage1_run_dir: Union[Path, None] = None
denoiser_run_dir: Union[Path, None] = None
|