SYMBOL INDEX (112 symbols across 15 files) FILE: bddm/loader/dataset.py class SpectrogramDataset (line 27) | class SpectrogramDataset(data.Dataset): method __init__ (line 29) | def __init__(self, data_dir, n_gpus, is_sampling, sampling_rate, method reset (line 78) | def reset(self): method find_all_wavs_in_dir (line 93) | def find_all_wavs_in_dir(self, data_dir): method find_all_mels_in_dir (line 107) | def find_all_mels_in_dir(self, data_dir): method crop_audio_and_mel (line 119) | def crop_audio_and_mel(self, audio, mel_spec): method __getitem__ (line 149) | def __getitem__(self, index): method __len__ (line 208) | def __len__(self): function create_train_and_valid_dataloader (line 220) | def create_train_and_valid_dataloader(config): function create_generation_dataloader (line 259) | def create_generation_dataloader(config): FILE: bddm/loader/stft.py class STFT (line 43) | class STFT(nn.Module): method __init__ (line 45) | def __init__(self, filter_length=800, hop_length=200, win_length=800, method transform (line 78) | def transform(self, input_data): method forward (line 108) | def forward(self, input_data): class TacotronSTFT (line 114) | class TacotronSTFT(nn.Module): method __init__ (line 118) | def __init__(self, filter_length=1024, hop_length=256, win_length=1024, method spectral_normalize (line 130) | def spectral_normalize(self, x): method spectral_de_normalize (line 133) | def spectral_de_normalize(self, x): method mel_spectrogram (line 136) | def mel_spectrogram(self, y): FILE: bddm/models/__init__.py function get_score_network (line 18) | def get_score_network(config): function get_schedule_network (line 25) | def get_schedule_network(config): FILE: bddm/models/diffwave.py function calc_diffusion_step_embedding (line 22) | def calc_diffusion_step_embedding(diffusion_steps, diffusion_step_embed_... function swish (line 55) | def swish(x): class Conv (line 61) | class Conv(nn.Module): method __init__ (line 62) | def __init__(self, in_channels, out_channels, kernel_size=3, dilation=1): method forward (line 70) | def forward(self, x): class ZeroConv1d (line 77) | class ZeroConv1d(nn.Module): method __init__ (line 78) | def __init__(self, in_channel, out_channel): method forward (line 84) | def forward(self, x): class ResidualBlock (line 91) | class ResidualBlock(nn.Module): method __init__ (line 92) | def __init__(self, res_channels, skip_channels, dilation, method forward (line 127) | def forward(self, input_data): class ResidualGroup (line 166) | class ResidualGroup(nn.Module): method __init__ (line 167) | def __init__(self, res_channels, skip_channels, num_res_layers, dilati... method forward (line 187) | def forward(self, input_data): class DiffWave (line 209) | class DiffWave(nn.Module): method __init__ (line 210) | def __init__(self, in_channels, res_channels, skip_channels, out_chann... method forward (line 231) | def forward(self, input_data): FILE: bddm/models/galr.py class Encoder (line 21) | class Encoder(nn.Module): method __init__ (line 23) | def __init__(self, enc_dim, win_len): method forward (line 38) | def forward(self, signals): class BiLSTMproj (line 54) | class BiLSTMproj(nn.Module): method __init__ (line 56) | def __init__(self, enc_dim, hid_dim): method forward (line 75) | def forward(self, intra_segs): class AttnPositionalEncoding (line 95) | class AttnPositionalEncoding(nn.Module): method __init__ (line 97) | def __init__(self, enc_dim, attn_max_len=5000): method forward (line 115) | def forward(self, x): class GlobalAttnLayer (line 128) | class GlobalAttnLayer(nn.Module): method __init__ (line 130) | def __init__(self, enc_dim, n_attn_head, attn_dropout): method forward (line 148) | def forward(self, inter_segs): class DeepGlobalAttnLayer (line 165) | class DeepGlobalAttnLayer(nn.Module): method __init__ (line 167) | def __init__(self, enc_dim, n_attn_head, attn_dropout, n_attn_layer=1): method forward (line 187) | def forward(self, inter_segs): class GALRBlock (line 205) | class GALRBlock(nn.Module): method __init__ (line 207) | def __init__(self, enc_dim, hid_dim, seg_len, low_dim=8, n_attn_head=8... method forward (line 234) | def forward(self, segments): class _GALR (line 279) | class _GALR(nn.Module): method __init__ (line 281) | def __init__(self, n_block, enc_dim, hid_dim, win_len, seg_len, method forward (line 317) | def forward(self, noisy_signals): method pad_zeros (line 338) | def pad_zeros(self, signals): method pad_segment (line 360) | def pad_segment(self, frames): method split_feature (line 383) | def split_feature(self, frames): class GALR (line 407) | class GALR(nn.Module): method __init__ (line 409) | def __init__(self, blocks=2, input_dim=128, hidden_dim=128, window_len... method forward (line 425) | def forward(self, audio, scales): FILE: bddm/sampler/sampler.py class Sampler (line 34) | class Sampler(object): method __init__ (line 38) | def __init__(self, config): method reset (line 72) | def reset(self): method draw_reference_data_pair (line 89) | def draw_reference_data_pair(self): method generate (line 96) | def generate(self): method noise_scheduling (line 128) | def noise_scheduling(self, ddim=False): method sampling (line 197) | def sampling(self, schedule=None, condition=None, method noise_scheduling_with_params (line 274) | def noise_scheduling_with_params(self, alpha_param, beta_param): method noise_scheduling_without_params (line 336) | def noise_scheduling_without_params(self): method assess (line 389) | def assess(self, generated_audio, audio_key=None): FILE: bddm/trainer/bmuf.py class BmufTrainer (line 21) | class BmufTrainer(object): method __init__ (line 23) | def __init__(self, master_node, rank, world_size, model, block_momentu... method check_all_processes_running (line 56) | def check_all_processes_running(self): method get_average_valid_loss (line 65) | def get_average_valid_loss(self, val_loss): method update_and_sync (line 83) | def update_and_sync(self, val_loss=None): method _copy_vec_to_param (line 116) | def _copy_vec_to_param(self, vec): FILE: bddm/trainer/ema.py class EMAHelper (line 16) | class EMAHelper(object): method __init__ (line 18) | def __init__(self, mu=0.999): method register (line 28) | def register(self, module): method update (line 41) | def update(self, module): method ema (line 55) | def ema(self, module): method ema_copy (line 68) | def ema_copy(self, module): method state_dict (line 87) | def state_dict(self): method load_state_dict (line 96) | def load_state_dict(self, state_dict): FILE: bddm/trainer/loss.py class ScoreLoss (line 17) | class ScoreLoss(nn.Module): method __init__ (line 19) | def __init__(self, config, diff_params): method forward (line 32) | def forward(self, model, mels, audios): class StepLoss (line 54) | class StepLoss(nn.Module): method __init__ (line 56) | def __init__(self, config, diff_params): method forward (line 70) | def forward(self, model, mels, audios): FILE: bddm/trainer/trainer.py class Trainer (line 31) | class Trainer(object): method __init__ (line 33) | def __init__(self, config): method reset (line 93) | def reset(self): method train (line 138) | def train(self): method _run_one_epoch (line 189) | def _run_one_epoch(self, validate=False): method serialize (line 251) | def serialize(self): FILE: bddm/utils/check_utils.py function check_score_network (line 18) | def check_score_network(config): function check_schedule_network (line 46) | def check_schedule_network(config): FILE: bddm/utils/diffusion_utils.py function compute_diffusion_params (line 16) | def compute_diffusion_params(beta): function map_noise_scale_to_time_step (line 39) | def map_noise_scale_to_time_step(alpha_infer, alpha): FILE: bddm/utils/log_utils.py function ctime (line 17) | def ctime(): function head (line 27) | def head(): function log (line 37) | def log(msg, config): FILE: egs/lj/main.py function dict_hash_5char (line 19) | def dict_hash_5char(dictionary): function start_exp (line 27) | def start_exp(config, config_hash): FILE: egs/vctk/main.py function dict_hash_5char (line 19) | def dict_hash_5char(dictionary): function start_exp (line 27) | def start_exp(config, config_hash):