Exemplo n.º 1
0
 def check_values(self,):
     """Check config fields"""
     c = asdict(self)
     check_argument("name", c, restricted=True)
     check_argument("path", c, restricted=True)
     check_argument("meta_file_train", c, restricted=True)
     check_argument("meta_file_val", c, restricted=False)
     check_argument("meta_file_attn_mask", c, restricted=False)
Exemplo n.º 2
0
 def check_values(
     self,
 ):
     """Check config fields"""
     c = asdict(self)
     check_argument("val_d", c, restricted=True, min_val=10, max_val=2056)
     check_argument("val_e", c, restricted=True, min_val=128, max_val=4058, allow_none=True)
     check_argument("val_f", c, restricted=True)
     check_argument("sc_list", c, restricted=True, allow_none=True)
     check_argument("sc", c, restricted=True, allow_none=True)
Exemplo n.º 3
0
 def check_values(
     self,
 ):
     """Check config fields"""
     c = asdict(self)
     check_argument("pad", c, prerequest="characters", restricted=True)
     check_argument("eos", c, prerequest="characters", restricted=True)
     check_argument("bos", c, prerequest="characters", restricted=True)
     check_argument("characters", c, prerequest="characters", restricted=True)
     check_argument("phonemes", c, restricted=True)
     check_argument("punctuations", c, prerequest="characters", restricted=True)
Exemplo n.º 4
0
 def check_values(
     self,
 ):
     """Check config fields"""
     c = asdict(self)
     super().check_values()
     check_argument("gst_style_input_weights", c, restricted=False)
     check_argument("gst_style_input_wav", c, restricted=False)
     check_argument("gst_embedding_dim", c, restricted=True, min_val=0, max_val=1000)
     check_argument("gst_use_speaker_embedding", c, restricted=False)
     check_argument("gst_num_heads", c, restricted=True, min_val=2, max_val=10)
     check_argument("gst_num_style_tokens", c, restricted=True, min_val=1, max_val=1000)
Exemplo n.º 5
0
    def check_values(self,):
        """Check config fields"""
        c = asdict(self)
        check_argument("num_mels", c, restricted=True, min_val=10, max_val=2056)
        check_argument("fft_size", c, restricted=True, min_val=128, max_val=4058)
        check_argument("sample_rate", c, restricted=True, min_val=512, max_val=100000)
        check_argument(
            "frame_length_ms",
            c,
            restricted=True,
            min_val=10,
            max_val=1000,
            alternative="win_length",
        )
        check_argument(
            "frame_shift_ms",
            c,
            restricted=True,
            min_val=1,
            max_val=1000,
            alternative="hop_length",
        )
        check_argument("preemphasis", c, restricted=True, min_val=0, max_val=1)
        check_argument("min_level_db", c, restricted=True, min_val=-1000, max_val=10)
        check_argument("ref_level_db", c, restricted=True, min_val=0, max_val=1000)
        check_argument("power", c, restricted=True, min_val=1, max_val=5)
        check_argument(
            "griffin_lim_iters", c, restricted=True, min_val=10, max_val=1000
        )

        # normalization parameters
        check_argument("signal_norm", c, restricted=True)
        check_argument("symmetric_norm", c, restricted=True)
        check_argument("max_norm", c, restricted=True, min_val=0.1, max_val=1000)
        check_argument("clip_norm", c, restricted=True)
        check_argument("mel_fmin", c, restricted=True, min_val=0.0, max_val=1000)
        check_argument("mel_fmax", c, restricted=True, min_val=500.0, allow_none=True)
        check_argument("spec_gain", c, restricted=True, min_val=1, max_val=100)
        check_argument("do_trim_silence", c, restricted=True)
        check_argument("trim_db", c, restricted=True)
Exemplo n.º 6
0
 def check_values(
     self,
 ):
     """Check config fields"""
     c = asdict(self)
     super().check_values()
     check_argument("capacitron_capacity", c, restricted=True, min_val=10, max_val=500)
     check_argument("capacitron_VAE_embedding_dim", c, restricted=True, min_val=16, max_val=1024)
     check_argument("capacitron_use_speaker_embedding", c, restricted=False)
     check_argument("capacitron_text_summary_embedding_dim", c, restricted=False, min_val=16, max_val=512)
     check_argument("capacitron_VAE_loss_alpha", c, restricted=False)
     check_argument("capacitron_grad_clip", c, restricted=False)