MCPcopy Create free account

hub / github.com/b04901014/FG-transformer-TTS / functions

Functions71 in github.com/b04901014/FG-transformer-TTS

↓ 12 callersMethod__init__
(self, nmels, hidden_size=512, dropout=0.2)
ETTS/module.py:44
↓ 6 callersFunction_make_guided_attention_mask
(ilen, rilen, olen, rolen, sigma)
ETTS/module.py:94
↓ 4 callersMethodinference
(self, emotion_global, emotion_local, text, maxlen=1000, threshold=.5, ref_wav=True)
ETTS/ettstransformer.py:104
↓ 4 callersMethodlength2mask
(self, x, length)
ETTS/ettstransformer.py:47
↓ 4 callersMethodplot_attn
(self, attns, prefix)
ETTS/trainer.py:198
↓ 3 callersMethodlength2mask
(self, x, length)
ETTS/baseline.py:42
↓ 2 callersFunctionRandomBucketSampler
(nbuckets, length, batch_size, drop_last, distributed=False, world_size=None, rank=Non
ETTS/dataloader.py:110
↓ 2 callersMethodsynthesize_with_ref
(self, text, global_ref_audio, local_ref_audio, outputname, wav_input=False)
ETTS/tester.py:53
↓ 2 callersMethodsynthesize_with_sample
(self, g_emo, text, outputname)
ETTS/tester.py:77
↓ 1 callersMethodget_feat_extract_output_lengths
Computes the output length of the convolutional layers
wav2vec2/FeatureFuser.py:95
↓ 1 callersFunctionprepare_mask
(length, shape, dtype, device)
wav2vec2/FeatureFuser.py:9
Method__getitem__
(self, i)
ETTS/dataloader.py:30
Method__getitem__
(self, i)
ETTS/dataloader.py:78
Method__init__
(self, text_embed_dim, emo_embed_dim, nmels, maxlength, symbols, ngst, d_model=512, hidden_si
ETTS/baseline.py:13
Method__init__
(self, text_embed_dim, emo_embed_dim, nmels, maxlength, ngst, nlst, model_dim, model_hidden_s
ETTS/tester.py:16
Method__init__
(self, datadir)
ETTS/dataloader.py:10
Method__init__
(self, datadir)
ETTS/dataloader.py:68
Method__init__
(self, nbuckets, length, batch_size, drop_last)
ETTS/dataloader.py:117
Method__init__
(self, nbuckets, length, batch_size, drop_last, num_replicas, rank, seed=787)
ETTS/dataloader.py:153
Method__init__
(self, text_embed_dim, emo_embed_dim, nmels, maxlength, symbols, ngst, nlst, d_model=512, hid
ETTS/ettstransformer.py:13
Method__init__
(self, ndim, hidden_size=256, dropout=0.2)
ETTS/module.py:8
Method__init__
(self, nmels, bottleneck_size=32, hidden_size=256)
ETTS/module.py:28
Method__init__
(self, maxlength, ndim, scaled=True)
ETTS/module.py:76
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1)
ETTS/module.py:105
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1)
ETTS/module.py:134
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1)
ETTS/module.py:180
Method__init__
(self, decoder_layers, guided_sigma=0.7, guided_layers=None, norm=None)
ETTS/module.py:214
Method__init__
(self, decoder_layer, num_layers, d_model, nhead, guided_sigma=0.5, guided_layers=None,
ETTS/module.py:256
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1)
ETTS/module.py:308
Method__init__
(self, encoder_layers, guided_sigma=0.3, guided_layers=None, norm=None)
ETTS/module.py:334
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1)
ETTS/module.py:365
Method__init__
(self, encoder_layers, guided_sigma=0.3, guided_layers=None, norm=None)
ETTS/module.py:408
Method__init__
(self, datadir, datatype, vocoder_ckpt_path, maxlength, nmels, emo_embed_dim, text_embed_dim,
ETTS/trainer.py:18
Method__init__
(self)
wav2vec2/wrapper.py:5
Method__init__
(self, pretrain=True)
wav2vec2/FeatureFuser.py:24
Method__iter__
(self)
ETTS/dataloader.py:129
Method__iter__
(self)
ETTS/dataloader.py:180
Method__len__
(self)
ETTS/dataloader.py:27
Method__len__
(self)
ETTS/dataloader.py:146
Method__len__
(self)
ETTS/dataloader.py:216
Method_conv_out_length
(input_length, kernel_size, stride)
wav2vec2/FeatureFuser.py:99
Methodconfigure_optimizers
(self)
ETTS/trainer.py:105
Methodforward
(self, emotion_global, text, mels, length_emo_global, length_text, length_mels)
ETTS/baseline.py:52
Methodforward
(self, emotion_global, emotion_local, text, mels, length_emo_global, length_emo_local, length_text, length_mel
ETTS/ettstransformer.py:57
Methodforward
(self, x, mask=None)
ETTS/module.py:14
Methodforward
(self, x, dropout)
ETTS/module.py:34
Methodforward
(self, x)
ETTS/module.py:66
Methodforward
(self, x)
ETTS/module.py:88
Methodforward
(self, tgt, memory, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None, memory_key_pad
ETTS/module.py:122
Methodforward
(self, tgt, text, emotion, tgt_mask=None, text_mask=None, emo_mask=None, tgt_key_padding_mask=
ETTS/module.py:156
Methodforward
(self, tgt, memory, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None, memory_key_pad
ETTS/module.py:198
Methodforward
(self, tgt, text, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None,
ETTS/module.py:224
Methodforward
(self, tgt, text, global_emotion, local_emotion, tgt_mask=None, tgt_key_padding_mask=None, tex
ETTS/module.py:270
Methodforward
(self, src, src_mask=None, src_key_padding_mask=None)
ETTS/module.py:323
Methodforward
(self, src, mask=None, src_key_padding_mask=None)
ETTS/module.py:344
Methodforward
(self, src, global_emotion, local_emotion, src_mask=None, src_key_padding_mask=None, emotion_k
ETTS/module.py:388
Methodforward
(self, src, global_emotion, local_emotion, mask=None, src_key_padding_mask=None, emotion_key_p
ETTS/module.py:418
Methodforward
(self, x, length=None)
wav2vec2/wrapper.py:9
Methodforward
(self, x, length=None)
wav2vec2/FeatureFuser.py:50
Methodinference
(self, emotion_global, text, maxlen=1000, threshold=.5, ref_wav=True)
ETTS/baseline.py:85
Methodlambda_lr
(current_step)
ETTS/trainer.py:112
Methodon_validation_epoch_end
(self)
ETTS/trainer.py:209
Methodon_validation_epoch_start
(self)
ETTS/trainer.py:154
MethodseqCollate
(self, batch)
ETTS/dataloader.py:46
MethodseqCollate
(self, batch)
ETTS/dataloader.py:86
Methodset_epoch
(self, epoch)
ETTS/dataloader.py:222
Methodtrain_dataloader
(self)
ETTS/trainer.py:81
Methodtrainable_params
(self)
wav2vec2/FeatureFuser.py:46
Methodtraining_step
(self, batch, batch_idx)
ETTS/trainer.py:126
Methodval_dataloader
(self)
ETTS/trainer.py:91
Methodvalidation_step
(self, batch, batch_idx)
ETTS/trainer.py:159