MCPcopy Create free account

hub / github.com/Audio-AGI/AudioSep / functions

Functions370 in github.com/Audio-AGI/AudioSep

Methodforward
(self, x: torch.Tensor)
models/CLAP/open_clip/model.py:90
Methodforward
(self, x)
models/CLAP/open_clip/model.py:120
Methodforward
(self, x)
models/CLAP/open_clip/model.py:233
Methodforward
(self, x: torch.Tensor)
models/CLAP/open_clip/model.py:247
Methodforward
(self, x: torch.Tensor)
models/CLAP/open_clip/model.py:255
Methodforward
(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
models/CLAP/open_clip/model.py:279
Methodforward
(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
models/CLAP/open_clip/model.py:299
Methodforward
(self, x: torch.Tensor)
models/CLAP/open_clip/model.py:346
Methodforward
Forward audio and text into the CLAP Parameters ---------- audio: torch.Tensor (batch_size, audio_length) the tim
models/CLAP/open_clip/model.py:682
Functionget_acc
(pred, target)
models/CLAP/open_clip/loss.py:343
Functionget_audioset632_id_to_lb
r"""Get AudioSet 632 classes ID to label mapping.
utils.py:75
Functionget_csv_dataset
(args, preprocess_fn, is_train)
models/CLAP/training/data.py:849
Functionget_data_from_log
Output dictionary from out.txt log file
models/CLAP/open_clip/utils.py:240
Functionget_imagenet
(args, preprocess_fns, split)
models/CLAP/training/data.py:310
Methodget_logit_scale
(self)
models/CLAP/open_clip/model.py:729
Functionget_map
(pred, target)
models/CLAP/open_clip/loss.py:337
Functionget_mauc
(pred, target)
models/CLAP/open_clip/loss.py:349
Methodget_median_metrics
(stats_dict, metric_type)
evaluation/evaluate_audioset.py:150
Functionget_ss_model
r"""Load trained universal source separation model. Args: configs (Dict) checkpoint_path (str): path of the checkpoint to load
utils.py:326
Functionget_tar_path_from_txts
Get tar path from txt path
models/CLAP/open_clip/utils.py:129
Functionget_toy_dataset
(args, model_cfg, is_train)
models/CLAP/training/data.py:878
Functionget_wds_dataset
Get a dataset for wdsdataloader.
models/CLAP/training/data.py:676
Functionids_to_hots
(ids, classes_num, device)
utils.py:141
Functioninit_embedding
Initialize a Linear or Convolutional layer.
models/base.py:24
Functioninit_gru
Initialize a GRU layer.
models/base.py:33
Functionint16_to_float32
(x: int)
utils.py:57
Functionis_using_horovod
()
models/CLAP/training/distributed.py:24
Functionl1_wav
(output_dict, target_dict)
losses.py:8
Functionlecun_normal_
(tensor)
models/CLAP/open_clip/htsat.py:314
Functionlinear_warm_up
r"""Get linear warm up scheduler for LambdaLR. Args: step (int): global step warm_up_steps (int): steps for warm up reduc
optimizers/lr_schedulers.py:5
Functionlist_pretrained
returns list of pretrained models Returns a tuple (model_name, pretrain_tag) by default or 'name:tag' if as_str == True
models/CLAP/open_clip/pretrained.py:78
Functionlist_pretrained_model_tags
return all pretrain tags for the specified model architecture
models/CLAP/open_clip/pretrained.py:98
Functionload_pretrained_panns
r"""Load pretrained pretrained audio neural networks (PANNs). Args: model_type: str, e.g., "Cnn14" checkpoint_path, str, e.g., "C
utils.py:89
Methodlock
lock modules Args: unlocked_groups (int): leave last n layer groups unlocked (default: 0)
models/CLAP/open_clip/timm_model.py:76
Methodlock
(self, unlocked_groups=0, freeze_bn_stats=False)
models/CLAP/open_clip/model.py:214
Methodlock
(self, unlocked_groups=0, freeze_bn_stats=False)
models/CLAP/open_clip/model.py:339
Functionlog_and_continue
Call in an exception handler to ignore any exception, isssue a warning, and continue.
models/CLAP/training/data.py:371
Functionloudness
Loudness normalize a signal. Normalize an input signal to a user loudness in dB LKFS. Params ------- data : torch.Tensor
utils.py:293
Functionmagnitude_to_db
(x)
utils.py:132
Methodno_weight_decay
(self)
models/CLAP/open_clip/htsat.py:1005
Methodno_weight_decay_keywords
(self)
models/CLAP/open_clip/htsat.py:1009
Methodon_train_batch_end
r"""Save a checkpoint every N steps.
callbacks/base.py:23
Functionpad_framewise_output
Pad framewise_output to the same length as input frames. The pad value is the same as the value of the last frame. Args: framewise_outpu
models/CLAP/open_clip/utils.py:202
Functionparse
(x)
models/CLAP/open_clip/htsat.py:31
Methodpredict
Run a single prediction on the model
predict.py:20
Methodprepare_data
(self)
data/datamodules.py:38
Functionpreprocess
Preprocess a single sample for wdsdataloader.
models/CLAP/training/data.py:566
Functionpreprocess_txt
(text)
models/CLAP/training/data.py:253
Functionprocess_ipc
(index_path, classes_num, filename)
models/CLAP/open_clip/utils.py:220
Functionrandom_loudness_norm
(audio, lower_db=-35, higher_db=-15, sr=32000)
data/waveform_mixers.py:112
Functionrandom_seed
(seed=42, rank=0)
models/CLAP/training/lp_main.py:130
Functionrandom_seed
(seed=42, rank=0)
models/CLAP/training/main.py:124
Functionremove_silence
r"""Remove silent frames.
utils.py:233
Functionrepeat_to_length
r"""Repeat audio to length.
utils.py:265
Methodrepeat_wat2img
(self, x, cur_pos)
models/CLAP/open_clip/htsat.py:1106
Functionsave_json
(data, name="data.json")
models/CLAP/open_clip/utils.py:304
Functionsave_p
(obj, filename)
models/CLAP/open_clip/utils.py:278
Methodsetup
Load the model into memory to make running multiple predictions efficient
predict.py:11
Methodsetup
r"""called on every device.
data/datamodules.py:43
Methodspectrogram_to_wav
Spectrogram to waveform. Args: input: (batch_size, segment_samples, channels_num) spectrogram: (batch_size, channels_num,
models/base.py:133
Methodteardown
(self)
data/datamodules.py:79
Methodtest_dataloader
(self)
data/datamodules.py:74
Methodtest_step
(self, batch, batch_idx)
models/audiosep.py:115
Methodtime_shifting
(self, x)
models/CLAP/training/data.py:114
Methodtrain_dataloader
r"""Get train loader.
data/datamodules.py:55
Methodtraining_step
r"""Forward a mini-batch data to model, calculate loss function, and train for one step. A mini-batch data is evenly distributed to multiple
models/audiosep.py:52
Methodval_dataloader
(self)
data/datamodules.py:69
Methodwav_to_spectrogram
Waveform to spectrogram. Args: input: (batch_size, segment_samples, channels_num) Outputs: output: (batch_size,
models/base.py:115
Functionwds_batch_list2dict
Return a dictionary of the batch, with keys as the names of the fields.
models/CLAP/training/data.py:827
Functionzero_shot_eval
(model, data, epoch, args)
models/CLAP/training/zero_shot.py:67
← previous301–370 of 370, ranked by callers