SYMBOL INDEX (461 symbols across 34 files)

FILE: audioset/dataset.py
  function default_config (line 26) | def default_config():
  function LMODE_default_config (line 48) | def LMODE_default_config():
  function decode_mp3 (line 55) | def decode_mp3(mp3_arr):
  function pad_or_truncate (line 73) | def pad_or_truncate(x, audio_length):
  function get_ir_sample (line 85) | def get_ir_sample(ir_path, _run, ir_augment, cut_irs_offset=None):
  function pydub_augment (line 104) | def pydub_augment(waveform, gain_augment=7, ir_augment=0):
  class MixupDataset (line 115) | class MixupDataset(TorchDataset):
    method __init__ (line 119) | def __init__(self, dataset, beta=2, rate=0.5):
    method __getitem__ (line 125) | def __getitem__(self, index):
    method __len__ (line 139) | def __len__(self):
  class AudioSetDataset (line 143) | class AudioSetDataset(TorchDataset):
    method __init__ (line 144) | def __init__(self, hdf5_file, sample_rate=32000, classes_num=527, clip...
    method open_hdf5 (line 164) | def open_hdf5(self):
    method __len__ (line 167) | def __len__(self):
    method __del__ (line 170) | def __del__(self):
    method __getitem__ (line 175) | def __getitem__(self, index):
    method resample (line 202) | def resample(self, waveform):
  function get_base_training_set (line 220) | def get_base_training_set(balanced_train_hdf5):
  function get_unbalanced_training_set (line 226) | def get_unbalanced_training_set(unbalanced_train_hdf5):
  function get_norms_dataset (line 233) | def get_norms_dataset(unbalanced_train_hdf5, balanced_train_hdf5):
  function get_base_full_training_set (line 240) | def get_base_full_training_set():
  function preload_mp3 (line 247) | def preload_mp3(balanced_train_hdf5, unbalanced_train_hdf5, num_of_class...
  function get_ft_cls_balanced_sample_weights (line 258) | def get_ft_cls_balanced_sample_weights(balanced_train_hdf5, unbalanced_t...
  function get_ft_weighted_sampler (line 294) | def get_ft_weighted_sampler(samples_weights=CMD(".get_ft_cls_balanced_sa...
  function get_base_test_set (line 310) | def get_base_test_set(eval_hdf5):
  function get_roll_func (line 316) | def get_roll_func(axis=1, shift=None, shift_range=50):
  function get_training_set (line 333) | def get_training_set(normalize, roll, wavmix=False):
  function get_full_training_set (line 349) | def get_full_training_set(normalize, roll, wavmix=False):
  function get_test_set (line 365) | def get_test_set(normalize):
  function print_conf (line 375) | def print_conf(_config):
  class DistributedSamplerWrapper (line 381) | class DistributedSamplerWrapper(DistributedSampler):
    method __init__ (line 382) | def __init__(
    method __iter__ (line 392) | def __iter__(self):
  function default_command (line 410) | def default_command():

FILE: audioset/prepare_scripts/convert_to_mp3.py
  function process_folder (line 15) | def process_folder(fol="balanced_train_segments"):
  function process_one (line 28) | def process_one(i, f1, f2):

FILE: audioset/prepare_scripts/create_h5pymp3_dataset.py
  function read_metadata (line 18) | def read_metadata(csv_path, classes_num, id_to_ix):
  function check_available (line 75) | def check_available(balanced_csv,balanced_audio_path,prefix=None):

FILE: ba3l/experiment.py
  function ingredients_recursive_apply (line 19) | def ingredients_recursive_apply(ing, fn):
  function config_recursive_apply (line 24) | def config_recursive_apply(conf, fn):
  function get_loggers (line 32) | def get_loggers(use_tensorboard_logger=False, use_sacred_logger=False):
  class Experiment (line 43) | class Experiment(Sacred_Experiment):
    method __init__ (line 48) | def __init__(
    method get_run_identifier (line 116) | def get_run_identifier(self):
    method get_dataloaders (line 121) | def get_dataloaders(self, filter={}):
    method get_train_dataloaders (line 130) | def get_train_dataloaders(self):
    method get_val_dataloaders (line 133) | def get_val_dataloaders(self):
    method _create_run (line 136) | def _create_run(
    method command (line 169) | def command(
    method add_default_args_config (line 205) | def add_default_args_config(self, function, prefix, extra_args={}, sta...

FILE: ba3l/ingredients/datasets.py
  function raise_ (line 14) | def raise_(ex):
  class Dataset (line 18) | class Dataset(Ingredient):
    method __init__ (line 29) | def __init__(
    method dataset (line 99) | def dataset(
    method iter (line 135) | def iter(
    method __getattr__ (line 174) | def __getattr__(self, k):
  class Datasets (line 183) | class Datasets(Ingredient, Munch):
    method get_instance (line 193) | def get_instance(cls):
    method __init__ (line 198) | def __init__(
    method __getattr__ (line 261) | def __getattr__(self, k):
    method __setattr__ (line 268) | def __setattr__(self, k, v):
    method __getitem__ (line 280) | def __getitem__(self, k):
    method __hash__ (line 294) | def __hash__(self):
    method get_datasets (line 297) | def get_datasets(self, config_conditions={}, return_datasets_names=Fal...

FILE: ba3l/ingredients/ingredient.py
  function raise_ (line 14) | def raise_(ex):
  class Ingredient (line 18) | class Ingredient(sacred_Ingredient):
    method __init__ (line 26) | def __init__(
    method add_default_args_config (line 84) | def add_default_args_config(self, function, prefix, extra_args={}, sta...
    method command (line 102) | def command(

FILE: ba3l/ingredients/models.py
  class Models (line 24) | class Models(Ingredient):
    method get_instance (line 34) | def get_instance(cls):
    method __init__ (line 39) | def __init__(
  function raise_ (line 98) | def raise_(ex):
  class Model (line 102) | class Model(Ingredient):
    method __init__ (line 112) | def __init__(
    method instance (line 176) | def instance(
    method __getattr__ (line 210) | def __getattr__(self, k):

FILE: ba3l/ingredients/trainer.py
  class Trainer (line 14) | class Trainer(Ingredient):

FILE: ba3l/module.py
  class Ba3lModule (line 33) | class Ba3lModule(pl.LightningModule):
    method __init__ (line 34) | def __init__(self, experiment):

FILE: ba3l/plutils/lr_monitor.py
  class LearningRateMonitor (line 30) | class LearningRateMonitor(Callback):
    method __init__ (line 70) | def __init__(self, logging_interval: Optional[str] = None, log_momentu...
    method on_train_start (line 79) | def on_train_start(self, trainer, *args, **kwargs):
    method on_train_batch_start (line 119) | def on_train_batch_start(self, trainer, *args, **kwargs):
    method on_train_epoch_start (line 130) | def on_train_epoch_start(self, trainer, *args, **kwargs):
    method _extract_stats (line 138) | def _extract_stats(self, trainer, interval: str) -> Dict[str, float]:
    method _extract_lr (line 158) | def _extract_lr(self, param_group, name: str) -> Dict[str, float]:
    method _extract_momentum (line 163) | def _extract_momentum(self, param_group, name: str, use_betas: bool) -...
    method _find_names (line 171) | def _find_names(self, lr_schedulers) -> List[str]:
    method _should_log (line 204) | def _should_log(trainer) -> bool:

FILE: ba3l/plutils/progress_bar.py
  class ProgressBar (line 17) | class ProgressBar(PlProgressBar):
    method init_validation_tqdm (line 66) | def init_validation_tqdm(self) -> tqdm:
    method on_epoch_start (line 83) | def on_epoch_start(self, trainer, pl_module):
    method on_train_batch_end (line 97) | def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch...
    method on_validation_start (line 103) | def on_validation_start(self, trainer, pl_module):
    method on_validation_batch_end (line 113) | def on_validation_batch_end(self, trainer, pl_module, outputs, batch, ...
    method on_validation_end (line 119) | def on_validation_end(self, trainer, pl_module):
  function convert_inf (line 127) | def convert_inf(x: Optional[Union[int, float]]) -> Optional[Union[int, f...
  function reset (line 134) | def reset(bar: tqdm, total: Optional[int] = None) -> None:

FILE: ba3l/util/functions.py
  function get_default_kwargs_dict (line 5) | def get_default_kwargs_dict(f):

FILE: ba3l/util/sacred_logger.py
  class SacredLogger (line 23) | class SacredLogger(LightningLoggerBase):
    method __init__ (line 24) | def __init__(self, sacred_experiment):
    method experiment (line 37) | def experiment(self):
    method run_id (line 41) | def run_id(self):
    method log_hyperparams (line 48) | def log_hyperparams(self, params):
    method log_metrics (line 53) | def log_metrics(self, metrics, step=None):
    method name (line 61) | def name(self):
    method version (line 65) | def version(self):
    method save (line 69) | def save(self):
    method finalize (line 76) | def finalize(self, status):

FILE: config_updates.py
  function add_configs (line 4) | def add_configs(ex):

FILE: esc50/dataset.py
  function default_config (line 29) | def default_config():
  function decode_mp3 (line 47) | def decode_mp3(mp3_arr):
  function pad_or_truncate (line 65) | def pad_or_truncate(x, audio_length):
  function get_ir_sample (line 77) | def get_ir_sample(ir_path, _run, ir_augment, cut_irs_offset=None):
  function pydub_augment (line 96) | def pydub_augment(waveform, gain_augment=7, ir_augment=0):
  class MixupDataset (line 107) | class MixupDataset(TorchDataset):
    method __init__ (line 111) | def __init__(self, dataset, beta=2, rate=0.5):
    method __getitem__ (line 117) | def __getitem__(self, index):
    method __len__ (line 131) | def __len__(self):
  class AudioSetDataset (line 137) | class AudioSetDataset(TorchDataset):
    method __init__ (line 138) | def __init__(self, meta_csv,  audiopath, fold, train=False, sample_rat...
    method __len__ (line 163) | def __len__(self):
    method __getitem__ (line 166) | def __getitem__(self, index):
    method resample (line 190) | def resample(self, waveform):
  function get_base_training_set (line 209) | def get_base_training_set(meta_csv, audio_path, fold=1):
  function get_ft_weighted_sampler (line 215) | def get_ft_weighted_sampler(samples_weights=CMD(".get_ft_cls_balanced_sa...
  function get_base_test_set (line 231) | def get_base_test_set(meta_csv, audio_path, fold=1):
  function get_roll_func (line 238) | def get_roll_func(axis=1, shift=None, shift_range=50):
  function get_training_set (line 255) | def get_training_set(normalize, roll, wavmix=False):
  function get_test_set (line 271) | def get_test_set(normalize):
  function print_conf (line 281) | def print_conf(_config):
  class DistributedSamplerWrapper (line 287) | class DistributedSamplerWrapper(DistributedSampler):
    method __init__ (line 288) | def __init__(
    method __iter__ (line 298) | def __iter__(self):
  function default_command (line 316) | def default_command():

FILE: ex_audioset.py
  function default_conf (line 52) | def default_conf():
  function get_scheduler_lambda (line 87) | def get_scheduler_lambda(warm_up_len=5, ramp_down_start=50, ramp_down_le...
  function get_lr_scheduler (line 98) | def get_lr_scheduler(optimizer, schedule_mode):
  function get_optimizer (line 105) | def get_optimizer(params, lr, adamw=True, weight_decay=0.0001):
  class M (line 112) | class M(Ba3lModule):
    method __init__ (line 113) | def __init__(self, experiment):
    method forward (line 139) | def forward(self, x):
    method mel_forward (line 142) | def mel_forward(self, x):
    method training_step (line 155) | def training_step(self, batch, batch_idx):
    method training_epoch_end (line 200) | def training_epoch_end(self, outputs):
    method predict (line 208) | def predict(self, batch, batch_idx: int, dataloader_idx: int = None):
    method validation_step (line 216) | def validation_step(self, batch, batch_idx):
    method validation_epoch_end (line 245) | def validation_epoch_end(self, outputs):
    method configure_optimizers (line 293) | def configure_optimizers(self):
    method configure_callbacks (line 304) | def configure_callbacks(self):
  function get_dynamic_norm (line 309) | def get_dynamic_norm(model, dyn_norm=False):
  function get_extra_checkpoint_callback (line 316) | def get_extra_checkpoint_callback(save_last_n=None):
  function get_extra_swa_callback (line 323) | def get_extra_swa_callback(swa=True, swa_epoch_start=50,
  function main (line 336) | def main(_run, _config, _log, _rnd, _seed, watch_model=True):
  function model_speed_test (line 365) | def model_speed_test(_run, _config, _log, _rnd, _seed, speed_test_batch_...
  function evaluate_only (line 430) | def evaluate_only(_run, _config, _log, _rnd, _seed):
  function test_loaders (line 445) | def test_loaders():
  function set_default_json_pickle (line 459) | def set_default_json_pickle(obj):
  function preload_mp3 (line 466) | def preload_mp3(all_y=CMD("/basedataset.preload_mp3")):
  function multiprocessing_run (line 475) | def multiprocessing_run(rank, word_size):
  function default_command (line 530) | def default_command():

FILE: ex_esc50.py
  function default_conf (line 50) | def default_conf():
  function get_scheduler_lambda (line 82) | def get_scheduler_lambda(warm_up_len=5, ramp_down_start=50, ramp_down_le...
  function get_lr_scheduler (line 92) | def get_lr_scheduler(optimizer, schedule_mode):
  function get_optimizer (line 99) | def get_optimizer(params, lr, adamw=True, weight_decay=0.0001):
  class M (line 106) | class M(Ba3lModule):
    method __init__ (line 107) | def __init__(self, experiment):
    method forward (line 126) | def forward(self, x):
    method mel_forward (line 129) | def mel_forward(self, x):
    method training_step (line 142) | def training_step(self, batch, batch_idx):
    method training_epoch_end (line 175) | def training_epoch_end(self, outputs):
    method predict (line 182) | def predict(self, batch, batch_idx: int, dataloader_idx: int = None):
    method validation_step (line 190) | def validation_step(self, batch, batch_idx):
    method validation_epoch_end (line 212) | def validation_epoch_end(self, outputs):
    method configure_optimizers (line 224) | def configure_optimizers(self):
    method configure_callbacks (line 235) | def configure_callbacks(self):
  function get_dynamic_norm (line 240) | def get_dynamic_norm(model, dyn_norm=False):
  function get_extra_checkpoint_callback (line 247) | def get_extra_checkpoint_callback(save_last_n=None):
  function get_extra_swa_callback (line 254) | def get_extra_swa_callback(swa=True, swa_epoch_start=2,
  function main (line 264) | def main(_run, _config, _log, _rnd, _seed):
  function model_speed_test (line 281) | def model_speed_test(_run, _config, _log, _rnd, _seed, speed_test_batch_...
  function evaluate_only (line 342) | def evaluate_only(_run, _config, _log, _rnd, _seed):
  function test_loaders (line 358) | def test_loaders():
  function set_default_json_pickle (line 372) | def set_default_json_pickle(obj):
  function multiprocessing_run (line 379) | def multiprocessing_run(rank, word_size):
  function default_command (line 431) | def default_command():

FILE: ex_fsd50k.py
  function variable_eval (line 54) | def variable_eval():
  function default_conf (line 60) | def default_conf():
  function get_scheduler_lambda (line 94) | def get_scheduler_lambda(warm_up_len=5, ramp_down_start=10, ramp_down_le...
  function get_lr_scheduler (line 104) | def get_lr_scheduler(optimizer, schedule_mode):
  function get_optimizer (line 111) | def get_optimizer(params, lr, adamw=True, weight_decay=0.0001):
  class M (line 118) | class M(Ba3lModule):
    method __init__ (line 119) | def __init__(self, experiment):
    method forward (line 138) | def forward(self, x):
    method mel_forward (line 141) | def mel_forward(self, x):
    method training_step (line 154) | def training_step(self, batch, batch_idx):
    method training_epoch_end (line 186) | def training_epoch_end(self, outputs):
    method predict (line 193) | def predict(self, batch, batch_idx: int, dataloader_idx: int = None):
    method validation_step (line 201) | def validation_step(self, batch, batch_idx, dataloader_idx):
    method validation_epoch_end (line 220) | def validation_epoch_end(self, outputs):
    method configure_optimizers (line 262) | def configure_optimizers(self):
    method configure_callbacks (line 273) | def configure_callbacks(self):
  function get_dynamic_norm (line 278) | def get_dynamic_norm(model, dyn_norm=False):
  function get_extra_checkpoint_callback (line 288) | def get_extra_checkpoint_callback(save_best=None):
  function get_extra_swa_callback (line 298) | def get_extra_swa_callback(swa=True, swa_epoch_start=10,
  function main (line 308) | def main(_run, _config, _log, _rnd, _seed):
  function model_speed_test (line 331) | def model_speed_test(_run, _config, _log, _rnd, _seed, speed_test_batch_...
  function evaluate_only (line 392) | def evaluate_only(_run, _config, _log, _rnd, _seed):
  function test_loaders (line 407) | def test_loaders():
  function set_default_json_pickle (line 421) | def set_default_json_pickle(obj):
  function preload_mp3 (line 428) | def preload_mp3(all_y=CMD("/basedataset.preload_mp3")):
  function multiprocessing_run (line 437) | def multiprocessing_run(rank, word_size):
  function default_command (line 489) | def default_command():

FILE: ex_openmic.py
  function default_conf (line 54) | def default_conf():
  function get_scheduler_lambda (line 88) | def get_scheduler_lambda(warm_up_len=5, ramp_down_start=50, ramp_down_le...
  function get_lr_scheduler (line 98) | def get_lr_scheduler(optimizer, schedule_mode):
  function get_optimizer (line 105) | def get_optimizer(params, lr, adamw=True, weight_decay=0.0001):
  class M (line 112) | class M(Ba3lModule):
    method __init__ (line 113) | def __init__(self, experiment):
    method forward (line 135) | def forward(self, x):
    method mel_forward (line 138) | def mel_forward(self, x):
    method training_step (line 151) | def training_step(self, batch, batch_idx):
    method training_epoch_end (line 190) | def training_epoch_end(self, outputs):
    method predict (line 197) | def predict(self, batch, batch_idx: int, dataloader_idx: int = None):
    method validation_step (line 205) | def validation_step(self, batch, batch_idx):
    method validation_epoch_end (line 228) | def validation_epoch_end(self, outputs):
    method configure_optimizers (line 272) | def configure_optimizers(self):
    method configure_callbacks (line 283) | def configure_callbacks(self):
  function get_dynamic_norm (line 288) | def get_dynamic_norm(model, dyn_norm=False):
  function get_extra_checkpoint_callback (line 295) | def get_extra_checkpoint_callback(save_last_n=None):
  function get_extra_swa_callback (line 302) | def get_extra_swa_callback(swa=True, swa_epoch_start=2,
  function main (line 312) | def main(_run, _config, _log, _rnd, _seed):
  function model_speed_test (line 330) | def model_speed_test(_run, _config, _log, _rnd, _seed, speed_test_batch_...
  function evaluate_only (line 391) | def evaluate_only(_run, _config, _log, _rnd, _seed):
  function test_loaders (line 406) | def test_loaders():
  function set_default_json_pickle (line 420) | def set_default_json_pickle(obj):
  function preload_mp3 (line 427) | def preload_mp3(all_y=CMD("/basedataset.preload_mp3")):
  function multiprocessing_run (line 436) | def multiprocessing_run(rank, word_size):
  function default_command (line 488) | def default_command():

FILE: fsd50k/dataset.py
  function default_config (line 25) | def default_config():
  function LMODE_default_config (line 48) | def LMODE_default_config():
  function decode_mp3 (line 52) | def decode_mp3(mp3_arr):
  function pad_or_truncate (line 70) | def pad_or_truncate(x, audio_length):
  function get_ir_sample (line 86) | def get_ir_sample(ir_path, _run, ir_augment, cut_irs_offset=None):
  function pydub_augment (line 105) | def pydub_augment(waveform, gain_augment=7, ir_augment=0):
  class MixupDataset (line 116) | class MixupDataset(TorchDataset):
    method __init__ (line 120) | def __init__(self, dataset, beta=2, rate=0.5):
    method __getitem__ (line 126) | def __getitem__(self, index):
    method __len__ (line 140) | def __len__(self):
  class AudioSetDataset (line 144) | class AudioSetDataset(TorchDataset):
    method __init__ (line 145) | def __init__(self, hdf5_file, sample_rate=32000, classes_num=200, clip...
    method open_hdf5 (line 167) | def open_hdf5(self):
    method __len__ (line 170) | def __len__(self):
    method __del__ (line 173) | def __del__(self):
    method __getitem__ (line 178) | def __getitem__(self, index):
    method resample (line 205) | def resample(self, waveform):
  function get_base_training_set (line 223) | def get_base_training_set(balanced_train_hdf5, clip_length=10):
  function preload_mp3 (line 229) | def preload_mp3(balanced_train_hdf5, unbalanced_train_hdf5, num_of_class...
  function get_ft_weighted_sampler (line 242) | def get_ft_weighted_sampler(samples_weights=CMD(".get_ft_cls_balanced_sa...
  function get_base_eval_set (line 258) | def get_base_eval_set(eval_hdf5, variable_eval=None):
  function get_base_valid_set (line 268) | def get_base_valid_set(valid_hdf5, variable_eval=None):
  function get_roll_func (line 278) | def get_roll_func(axis=1, shift=None, shift_range=50):
  function get_training_set (line 295) | def get_training_set(normalize, roll, wavmix=False):
  function get_valid_set (line 311) | def get_valid_set(normalize):
  function get_eval_set (line 321) | def get_eval_set(normalize):
  function print_conf (line 331) | def print_conf(_config):
  class DistributedSamplerWrapper (line 337) | class DistributedSamplerWrapper(DistributedSampler):
    method __init__ (line 338) | def __init__(
    method __iter__ (line 348) | def __iter__(self):
  function default_command (line 366) | def default_command():

FILE: fsd50k/prepare_scripts/convert_to_mp3.py
  function process_folder (line 19) | def process_folder(fol="balanced_train_segments"):
  function process_one (line 32) | def process_one(i, f1, f2):

FILE: fsd50k/prepare_scripts/create_h5pymp3_dataset.py
  function get_labels (line 49) | def get_labels(df):

FILE: helpers/audiodatasets.py
  function h6 (line 12) | def h6(w):
  class AudioPreprocessDataset (line 15) | class AudioPreprocessDataset(Dataset):
    method __init__ (line 23) | def __init__(self, files, labels, label_encoder, base_dir, preprocesso...
    method __getitem__ (line 36) | def __getitem__(self, index):
    method get_ordered_ids (line 42) | def get_ordered_ids(self):
    method get_ordered_labels (line 45) | def get_ordered_labels(self):
    method __len__ (line 48) | def __len__(self):
  class ObjectCacher (line 51) | class ObjectCacher:
    method __init__ (line 52) | def __init__(self, get_obj_func, dataset_name, obj_name="",
    method get_obj_cache_path (line 91) | def get_obj_cache_path(self):
    method get (line 94) | def get(self):
  class PreprocessDataset (line 99) | class PreprocessDataset(Dataset):
    method __init__ (line 106) | def __init__(self, dataset, preprocessor):
    method __getitem__ (line 112) | def __getitem__(self, index):
    method __len__ (line 114) | def __len__(self):
  class FilesCachedDataset (line 119) | class FilesCachedDataset(Dataset):
    method __init__ (line 120) | def __init__(self, get_dataset_func, dataset_name, x_name="",
    method __getitem__ (line 148) | def __getitem__(self, index):
    method get_ordered_labels (line 157) | def get_ordered_labels(self):
    method get_ordered_ids (line 160) | def get_ordered_ids(self):
    method get_xcache_path (line 163) | def get_xcache_path(self):
    method get_ycache_path (line 166) | def get_ycache_path(self):
    method get_sidcache_path (line 169) | def get_sidcache_path(self):
    method __len__ (line 172) | def __len__(self):
  class SelectionDataset (line 177) | class SelectionDataset(Dataset):
    method __init__ (line 184) | def __init__(self, dataset, sample_ids):
    method reselect (line 190) | def reselect(self, sample_ids):
    method get_ordered_ids (line 194) | def get_ordered_ids(self):
    method get_ordered_labels (line 197) | def get_ordered_labels(self):
    method __getitem__ (line 202) | def __getitem__(self, index):
    method __len__ (line 205) | def __len__(self):
  class SimpleSelectionDataset (line 208) | class SimpleSelectionDataset(Dataset):
    method __init__ (line 215) | def __init__(self, dataset, available_indexes ):
    method __getitem__ (line 219) | def __getitem__(self, index):
    method __len__ (line 222) | def __len__(self):

FILE: helpers/mixup.py
  function my_mixup (line 5) | def my_mixup(size, alpha):

FILE: helpers/models_size.py
  function count_non_zero_params (line 7) | def count_non_zero_params(model):

FILE: helpers/ramp.py
  function pseudo_rampup (line 8) | def pseudo_rampup(T1, T2):
  function exp_rampup (line 21) | def exp_rampup(rampup_length):
  function linear_rampup (line 33) | def linear_rampup(rampup_length):
  function linear_rampdown (line 45) | def linear_rampdown(rampdown_length, start=0, last_value=0):
  function exp_rampdown (line 57) | def exp_rampdown(rampdown_length, num_epochs):
  function cosine_rampdown (line 70) | def cosine_rampdown(rampdown_length, num_epochs):
  function exp_warmup (line 83) | def exp_warmup(rampup_length, rampdown_length, num_epochs):
  function exp_warmup_linear_down (line 93) | def exp_warmup_linear_down(warmup, rampdown_length, start_rampdown, last...
  function test_warmup (line 101) | def test_warmup():
  function test_warmupl (line 107) | def test_warmupl():
  function cosine_cycle (line 113) | def cosine_cycle(cycle_len=20,ramp_down_start=100,last_lr_value=0.01):

FILE: helpers/swa_callback.py
  class StochasticWeightAveraging (line 35) | class StochasticWeightAveraging(Callback):
    method __init__ (line 37) | def __init__(
    method swa_start (line 127) | def swa_start(self) -> int:
    method swa_end (line 131) | def swa_end(self) -> int:
    method pl_module_contains_batch_norm (line 135) | def pl_module_contains_batch_norm(pl_module: 'pl.LightningModule'):
    method setup (line 138) | def setup(self, trainer: 'pl.Trainer', pl_module: 'pl.LightningModule'...
    method on_fit_start (line 142) | def on_fit_start(self, trainer: 'pl.Trainer', pl_module: 'pl.Lightning...
    method on_train_epoch_start (line 161) | def on_train_epoch_start(self, trainer: 'pl.Trainer', pl_module: 'pl.L...
    method on_train_epoch_end (line 200) | def on_train_epoch_end(self, trainer: 'pl.Trainer',pl_module: 'pl.Ligh...
    method on_train_end (line 204) | def on_train_end(self, trainer: 'pl.Trainer', pl_module: 'pl.Lightning...
    method on_validation_epoch_start (line 208) | def on_validation_epoch_start(self, trainer, pl_module) -> None:
    method transfer_weights (line 217) | def transfer_weights(src_pl_module: 'pl.LightningModule', dst_pl_modul...
    method reset_batch_norm_and_save_state (line 221) | def reset_batch_norm_and_save_state(self, pl_module):
    method reset_momenta (line 239) | def reset_momenta(self):
    method update_parameters (line 247) | def update_parameters(
    method avg_fn (line 262) | def avg_fn(

FILE: helpers/swa_legacy.py
  class StochasticWeightAveraging (line 37) | class StochasticWeightAveraging(Callback):
    method __init__ (line 39) | def __init__(
    method swa_start (line 132) | def swa_start(self) -> int:
    method swa_end (line 136) | def swa_end(self) -> int:
    method pl_module_contains_batch_norm (line 140) | def pl_module_contains_batch_norm(pl_module: 'pl.LightningModule'):
    method on_before_accelerator_backend_setup (line 143) | def on_before_accelerator_backend_setup(self, trainer: 'pl.Trainer', p...
    method on_fit_start (line 147) | def on_fit_start(self, trainer: 'pl.Trainer', pl_module: 'pl.Lightning...
    method on_train_epoch_start (line 172) | def on_train_epoch_start(self, trainer: 'pl.Trainer', pl_module: 'pl.L...
    method on_train_epoch_end (line 211) | def on_train_epoch_end(self, trainer: 'pl.Trainer', pl_module: 'pl.Lig...
    method on_train_end (line 214) | def on_train_end(self, trainer: 'pl.Trainer', pl_module: 'pl.Lightning...
    method on_validation_epoch_start (line 217) | def on_validation_epoch_start(self, trainer, pl_module) -> None:
    method transfer_weights (line 225) | def transfer_weights(src_pl_module: 'pl.LightningModule', dst_pl_modul...
    method reset_batch_norm_and_save_state (line 229) | def reset_batch_norm_and_save_state(self, pl_module):
    method reset_momenta (line 247) | def reset_momenta(self):
    method update_parameters (line 255) | def update_parameters(
    method avg_fn (line 271) | def avg_fn(

FILE: helpers/workersinit.py
  function worker_init_fn (line 6) | def worker_init_fn(x):

FILE: models/helpers/vit_helpers.py
  function adapt_input_conv (line 27) | def adapt_input_conv(in_chans, conv_weight):
  function load_pretrained (line 54) | def load_pretrained(
  function overlay_external_default_cfg (line 144) | def overlay_external_default_cfg(default_cfg, kwargs):
  function filter_kwargs (line 153) | def filter_kwargs(kwargs, names):
  function set_default_kwargs (line 160) | def set_default_kwargs(kwargs, names, default_cfg):
  function update_default_cfg_and_kwargs (line 180) | def update_default_cfg_and_kwargs(default_cfg, kwargs, kwargs_filter):
  function drop_path (line 203) | def drop_path(x, drop_prob: float = 0.0, training: bool = False):
  class DropPath (line 225) | class DropPath(nn.Module):
    method __init__ (line 228) | def __init__(self, drop_prob=None):
    method forward (line 232) | def forward(self, x):
  function _no_grad_trunc_normal_ (line 239) | def _no_grad_trunc_normal_(tensor, mean, std, a, b):
  function trunc_normal_ (line 277) | def trunc_normal_(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0):
  function variance_scaling_ (line 297) | def variance_scaling_(tensor, scale=1.0, mode="fan_in", distribution="no...
  function lecun_normal_ (line 320) | def lecun_normal_(tensor):
  function build_model_with_cfg (line 324) | def build_model_with_cfg(

FILE: models/passt.py
  function _ntuple (line 26) | def _ntuple(n):
  function _cfg (line 42) | def _cfg(url='', **kwargs):
  function adapt_input_conv (line 246) | def adapt_input_conv(in_chans, conv_weight):
  class Mlp (line 271) | class Mlp(nn.Module):
    method __init__ (line 275) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 284) | def forward(self, x):
  class PatchEmbed (line 298) | class PatchEmbed(nn.Module):
    method __init__ (line 302) | def __init__(self, img_size=224, patch_size=16, stride=16, in_chans=3,...
    method forward (line 318) | def forward(self, x):
  class Attention (line 331) | class Attention(nn.Module):
    method __init__ (line 332) | def __init__(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., pro...
    method forward (line 343) | def forward(self, x):
  class Block (line 364) | class Block(nn.Module):
    method __init__ (line 366) | def __init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=...
    method forward (line 377) | def forward(self, x):
  class PaSST (line 383) | class PaSST(nn.Module):
    method __init__ (line 391) | def __init__(self, u_patchout=0, s_patchout_t=0, s_patchout_f=0, img_s...
    method init_weights (line 471) | def init_weights(self, mode=''):
    method _init_weights (line 486) | def _init_weights(self, m):
    method no_weight_decay (line 491) | def no_weight_decay(self):
    method get_classifier (line 494) | def get_classifier(self):
    method reset_classifier (line 500) | def reset_classifier(self, num_classes, global_pool=''):
    method forward_features (line 506) | def forward_features(self, x):
    method forward (line 576) | def forward(self, x):
  function _init_vit_weights (line 598) | def _init_vit_weights(module: nn.Module, name: str = '', head_bias: floa...
  function resize_pos_embed (line 633) | def resize_pos_embed(posemb, posemb_new, num_tokens=1, gs_new=(), mode='...
  function adapt_image_pos_embed_to_passt (line 656) | def adapt_image_pos_embed_to_passt(posemb, num_tokens=1, gs_new=(), mode...
  function checkpoint_filter_fn (line 679) | def checkpoint_filter_fn(state_dict, model):
  function _create_vision_transformer (line 709) | def _create_vision_transformer(variant, pretrained=False, default_cfg=No...
  function vit_huge_patch14_224_in21k (line 734) | def vit_huge_patch14_224_in21k(pretrained=False, **kwargs):
  function deit_base_distilled_patch16_384 (line 745) | def deit_base_distilled_patch16_384(pretrained=False, **kwargs):
  function passt_s_swa_p16_128_ap476 (line 756) | def passt_s_swa_p16_128_ap476(pretrained=False, **kwargs):
  function passt_s_kd_p16_128_ap486 (line 769) | def passt_s_kd_p16_128_ap486(pretrained=False, **kwargs):
  function passt_l_kd_p16_128_ap47 (line 782) | def passt_l_kd_p16_128_ap47(pretrained=False, **kwargs):
  function passt_s_swa_p16_128_ap4761 (line 795) | def passt_s_swa_p16_128_ap4761(pretrained=False, **kwargs):
  function passt_s_p16_128_ap472 (line 808) | def passt_s_p16_128_ap472(pretrained=False, **kwargs):
  function passt_s_p16_s12_128_ap470 (line 821) | def passt_s_p16_s12_128_ap470(pretrained=False, **kwargs):
  function passt_s_f128_20sec_p16_s10_ap474_swa (line 834) | def passt_s_f128_20sec_p16_s10_ap474_swa(pretrained=False, **kwargs):
  function passt_s_f128_30sec_p16_s10_ap473_swa (line 842) | def passt_s_f128_30sec_p16_s10_ap473_swa(pretrained=False, **kwargs):
  function passt_s_swa_p16_s12_128_ap473 (line 850) | def passt_s_swa_p16_s12_128_ap473(pretrained=False, **kwargs):
  function passt_s_p16_s14_128_ap469 (line 863) | def passt_s_p16_s14_128_ap469(pretrained=False, **kwargs):
  function passt_s_swa_p16_s14_128_ap471 (line 876) | def passt_s_swa_p16_s14_128_ap471(pretrained=False, **kwargs):
  function passt_s_swa_p16_s16_128_ap473 (line 889) | def passt_s_swa_p16_s16_128_ap473(pretrained=False, **kwargs):
  function passt_s_p16_s16_128_ap468 (line 902) | def passt_s_p16_s16_128_ap468(pretrained=False, **kwargs):
  function fix_embedding_layer (line 923) | def fix_embedding_layer(model, embed="default"):
  function lighten_model (line 933) | def lighten_model(model, cut_depth=0):
  function get_model (line 958) | def get_model(arch="passt_s_kd_p16_128_ap486", pretrained=True, n_classe...
  class EnsembelerModel (line 1021) | class EnsembelerModel(nn.Module):
    method __init__ (line 1022) | def __init__(self, models):
    method forward (line 1026) | def forward(self, x):
  function get_ensemble_model (line 1040) | def get_ensemble_model(arch_list=[]):

FILE: models/preprocess.py
  class AugmentMelSTFT (line 19) | class AugmentMelSTFT(nn.Module):
    method __init__ (line 20) | def __init__(self, n_mels=128, sr=32000, win_length=800, hopsize=320, ...
    method forward (line 57) | def forward(self, x):
    method extra_repr (line 88) | def extra_repr(self):

FILE: openmic/dataset.py
  function default_config (line 29) | def default_config():
  function decode_mp3 (line 47) | def decode_mp3(mp3_arr):
  function pad_or_truncate (line 65) | def pad_or_truncate(x, audio_length):
  function get_ir_sample (line 77) | def get_ir_sample(ir_path, _run, ir_augment, cut_irs_offset=None):
  function pydub_augment (line 96) | def pydub_augment(waveform, gain_augment=7, ir_augment=0):
  class MixupDataset (line 107) | class MixupDataset(TorchDataset):
    method __init__ (line 111) | def __init__(self, dataset, beta=2, rate=0.5):
    method __getitem__ (line 117) | def __getitem__(self, index):
    method __len__ (line 140) | def __len__(self):
  class AudioSetDataset (line 145) | class AudioSetDataset(TorchDataset):
    method __init__ (line 146) | def __init__(self, hdf5_file, sample_rate=32000, classes_num=527, clip...
    method open_hdf5 (line 166) | def open_hdf5(self):
    method __len__ (line 169) | def __len__(self):
    method __del__ (line 172) | def __del__(self):
    method __getitem__ (line 177) | def __getitem__(self, index):
    method resample (line 203) | def resample(self, waveform):
  function get_base_training_set (line 221) | def get_base_training_set(openmic_train_hdf5):
  function get_ft_weighted_sampler (line 227) | def get_ft_weighted_sampler(samples_weights=CMD(".get_ft_cls_balanced_sa...
  function get_base_test_set (line 243) | def get_base_test_set(openmic_test_hdf5):
  function get_roll_func (line 249) | def get_roll_func(axis=1, shift=None, shift_range=50):
  function get_training_set (line 266) | def get_training_set(normalize, roll, wavmix=False):
  function get_test_set (line 282) | def get_test_set(normalize):
  function print_conf (line 292) | def print_conf(_config):
  class DistributedSamplerWrapper (line 298) | class DistributedSamplerWrapper(DistributedSampler):
    method __init__ (line 299) | def __init__(
    method __iter__ (line 309) | def __iter__(self):
  function default_command (line 327) | def default_command():

FILE: openmic/prepare_scripts/download_preprocess.py
  function download (line 21) | def download(force=False):
  function untar (line 29) | def untar():
  function process_folder (line 36) | def process_folder(fol="balanced_train_segments"):
  function process_one (line 50) | def process_one(i, f1, f2):
  function make_mp3 (line 56) | def make_mp3():
  function read_metadata (line 60) | def read_metadata(csv_path, classes_num, id_to_ix, openmicf):
  function get_files_labels (line 95) | def get_files_labels(balanced_csv, balanced_audio_path, d_files, openmic...
  function pack (line 118) | def pack():
  function preprocess (line 164) | def preprocess():