SYMBOL INDEX (102 symbols across 20 files) FILE: neural_waveshaping_synthesis/data/general.py class GeneralDataset (line 9) | class GeneralDataset(torch.utils.data.Dataset): method __init__ (line 10) | def __init__(self, path: str, split: str = "train", load_to_memory: bo... method __len__ (line 34) | def __len__(self): method __getitem__ (line 37) | def __getitem__(self, idx): class GeneralDataModule (line 61) | class GeneralDataModule(pl.LightningDataModule): method __init__ (line 62) | def __init__( method prepare_data (line 75) | def prepare_data(self): method setup (line 78) | def setup(self, stage: str = None): method _make_dataloader (line 85) | def _make_dataloader(self, dataset): method train_dataloader (line 90) | def train_dataloader(self): method val_dataloader (line 93) | def val_dataloader(self): method test_dataloader (line 96) | def test_dataloader(self): FILE: neural_waveshaping_synthesis/data/urmp.py class URMPDataModule (line 9) | class URMPDataModule(GeneralDataModule): method __init__ (line 10) | def __init__( FILE: neural_waveshaping_synthesis/data/utils/create_dataset.py function create_directory (line 13) | def create_directory(path): function create_directories (line 25) | def create_directories(target_root, names): function make_splits (line 31) | def make_splits( function lazy_create_dataset (line 73) | def lazy_create_dataset( function create_dataset (line 152) | def create_dataset( FILE: neural_waveshaping_synthesis/data/utils/f0_extraction.py function extract_f0_with_crepe (line 17) | def extract_f0_with_crepe( function extract_f0_with_pyin (line 61) | def extract_f0_with_pyin( FILE: neural_waveshaping_synthesis/data/utils/loudness_extraction.py function compute_power_spectrogram (line 11) | def compute_power_spectrogram( function perform_perceptual_weighting (line 26) | def perform_perceptual_weighting( function extract_perceptual_loudness (line 43) | def extract_perceptual_loudness( function extract_rms (line 72) | def extract_rms( FILE: neural_waveshaping_synthesis/data/utils/mfcc_extraction.py function extract_mfcc (line 7) | def extract_mfcc( FILE: neural_waveshaping_synthesis/data/utils/preprocess_audio.py function read_audio_files (line 16) | def read_audio_files(files: list): function convert_to_float32_audio (line 21) | def convert_to_float32_audio(audio: np.ndarray): function make_monophonic (line 30) | def make_monophonic(audio: np.ndarray, strategy: str = "keep_left"): function normalise_signal (line 61) | def normalise_signal(audio: np.ndarray, factor: float): function resample_audio (line 65) | def resample_audio(audio: np.ndarray, original_sr: float, target_sr: flo... function segment_signal (line 69) | def segment_signal( function filter_segments (line 83) | def filter_segments( function preprocess_single_audio_file (line 96) | def preprocess_single_audio_file( function preprocess_audio (line 203) | def preprocess_audio( FILE: neural_waveshaping_synthesis/data/utils/upsampling.py function get_padded_length (line 9) | def get_padded_length(frames: int, window_length: int, hop_length: int): function get_source_target_axes (line 13) | def get_source_target_axes(frames: int, window_length: int, hop_length: ... function linear_interpolation (line 21) | def linear_interpolation( function cubic_spline_interpolation (line 38) | def cubic_spline_interpolation( function overlap_add_upsample (line 56) | def overlap_add_upsample( FILE: neural_waveshaping_synthesis/models/modules/dynamic.py class FiLM (line 6) | class FiLM(nn.Module): method forward (line 7) | def forward(self, x, gamma, beta): class TimeDistributedLayerNorm (line 11) | class TimeDistributedLayerNorm(nn.Module): method __init__ (line 12) | def __init__(self, size: int): method forward (line 16) | def forward(self, x): class TimeDistributedMLP (line 21) | class TimeDistributedMLP(nn.Module): method __init__ (line 22) | def __init__(self, in_size: int, hidden_size: int, out_size: int, dept... method forward (line 39) | def forward(self, x): FILE: neural_waveshaping_synthesis/models/modules/generators.py class FIRNoiseSynth (line 12) | class FIRNoiseSynth(nn.Module): method __init__ (line 13) | def __init__( method forward (line 21) | def forward(self, H_re): class HarmonicOscillator (line 39) | class HarmonicOscillator(nn.Module): method __init__ (line 40) | def __init__(self, n_harmonics, sample_rate): method _create_harmonic_axis (line 47) | def _create_harmonic_axis(self, n_harmonics): method _create_antialias_mask (line 50) | def _create_antialias_mask(self, f0): method _create_phase_shift (line 54) | def _create_phase_shift(self, n_harmonics): method forward (line 58) | def forward(self, f0): FILE: neural_waveshaping_synthesis/models/modules/shaping.py class Sine (line 10) | class Sine(nn.Module): method forward (line 11) | def forward(self, x: torch.Tensor): class TrainableNonlinearity (line 16) | class TrainableNonlinearity(nn.Module): method __init__ (line 17) | def __init__( method forward (line 36) | def forward(self, x): class NEWT (line 41) | class NEWT(nn.Module): method __init__ (line 42) | def __init__( method forward (line 67) | def forward(self, exciter, control_embedding): class FastNEWT (line 82) | class FastNEWT(NEWT): method __init__ (line 83) | def __init__( method _init_lookup_table (line 107) | def _init_lookup_table( method _lookup (line 121) | def _lookup(self, idx): method shaping_fn (line 136) | def shaping_fn(self, x): class Reverb (line 155) | class Reverb(nn.Module): method __init__ (line 156) | def __init__(self, length_in_seconds, sr): method forward (line 161) | def forward(self, x): FILE: neural_waveshaping_synthesis/models/neural_waveshaping.py class ControlModule (line 18) | class ControlModule(nn.Module): method __init__ (line 19) | def __init__(self, control_size: int, hidden_size: int, embedding_size... method forward (line 24) | def forward(self, x): class NeuralWaveshaping (line 30) | class NeuralWaveshaping(pl.LightningModule): method __init__ (line 31) | def __init__( method render_exciter (line 64) | def render_exciter(self, f0): method get_embedding (line 69) | def get_embedding(self, control): method forward (line 74) | def forward(self, f0, control): method configure_optimizers (line 92) | def configure_optimizers(self): method _run_step (line 104) | def _run_step(self, batch): method _log_audio (line 114) | def _log_audio(self, name, audio): method training_step (line 123) | def training_step(self, batch, batch_idx): method validation_step (line 136) | def validation_step(self, batch, batch_idx): method test_step (line 152) | def test_step(self, batch, batch_idx): FILE: neural_waveshaping_synthesis/utils/seed_all.py function seed_all (line 6) | def seed_all(seed): FILE: neural_waveshaping_synthesis/utils/utils.py function apply (line 5) | def apply(fn: Callable[[any], any], x: Sequence[any]): function apply_unpack (line 11) | def apply_unpack(fn: Callable[[any], any], x: Sequence[Sequence[any]]): function unzip (line 17) | def unzip(x: Sequence[any]): function make_dir_if_not_exists (line 21) | def make_dir_if_not_exists(path): FILE: scripts/create_dataset.py function get_filenames (line 10) | def get_filenames(directory): function main (line 20) | def main(gin_file, data_directory, output_directory, seed=0, device="cpu"): FILE: scripts/create_urmp_dataset.py function get_instrument_file_list (line 26) | def get_instrument_file_list(instrument_string, directory): function main (line 41) | def main(gin_file, data_directory, output_directory, seed=0, device="cpu"): FILE: scripts/resynthesise_dataset.py function main (line 26) | def main( FILE: scripts/time_buffer_sizes.py function main (line 24) | def main( FILE: scripts/time_forward_pass.py function main (line 23) | def main( FILE: scripts/train.py function get_model (line 11) | def get_model(model, with_wandb): function trainer_kwargs (line 16) | def trainer_kwargs(**kwargs): function main (line 29) | def main(