MCPcopy Create free account

hub / github.com/RetroCirce/HTS-Audio-Transformer / functions

Functions248 in github.com/RetroCirce/HTS-Audio-Transformer

⨍ Functions 248◇ Types & classes 59
Method__getitem__
Load waveform and target of an audio clip. Args: index: the index number Return: { "audio_name": str,
data_generator.py:287
Method__init__
Args: index_path: the link to each audio idc: npy file, the number of samples in each class, computed in main
data_generator.py:22
Method__init__
(self, dataset, config, eval_mode = False)
data_generator.py:186
Method__init__
(self, dataset, config, eval_mode = False)
data_generator.py:234
Method__init__
(self, dataset, config)
data_generator.py:281
Method__init__
(self, gamma_neg=4, gamma_pos=1, clip=0.05, eps=1e-8, disable_torch_grad_focal_loss=True)
utils.py:29
Method__init__
(self, in_channels, out_channels)
models.py:27
Method__init__
(self, in_channels, out_channels)
models.py:73
Method__init__
(self, n_in, n_out, activation='linear', temperature=1.)
models.py:110
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:141
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:232
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:316
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:486
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
models.py:584
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
models.py:642
Method__init__
(self, block, layers, zero_init_residual=False, groups=1, width_per_group=64, replace_stride_
models.py:699
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:773
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:860
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:947
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1034
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1125
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1216
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1307
Method__init__
(self, inp, oup, stride, expand_ratio)
models.py:1425
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1477
Method__init__
(self, in_channels, out_channels, kernel_size, stride)
models.py:1607
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1632
Method__init__
(self, in_channels, out_channels, kernel_size, stride)
models.py:1698
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1732
Method__init__
(self, in_channels, out_channels, kernel_size)
models.py:1805
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:1871
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
models.py:1947
Method__init__
(self, block, layers, zero_init_residual=False, groups=1, width_per_group=64, replace_stride_
models.py:2003
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2081
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2136
Method__init__
(self, in_channels, out_channels)
models.py:2191
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2228
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2316
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2430
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2544
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2641
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2738
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2831
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:2922
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:3014
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:3112
Method__init__
(self, sample_rate, window_size, hop_size, mel_bins, fmin, fmax, classes_num)
models.py:3217
Method__init__
(self, sed_models, config, dataset)
sed_model.py:260
Method__init__
(self, train_dataset, eval_dataset, device_num)
main.py:42
Method__init__
(self, drop_prob=None)
model/layers.py:55
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768, norm_layer=None, flatten=True, patch_stride = 1
model/layers.py:65
Method__init__
(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
model/htsat.py:73
Method__init__
(self, dim, input_resolution, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=
model/htsat.py:163
Method__init__
(self, input_resolution, dim, norm_layer=nn.LayerNorm)
model/htsat.py:277
Method__init__
(self, spec_size=256, patch_size=4, patch_stride=(4,4), in_chans=1, num_classes=527,
model/htsat.py:405
Method__len__
(self)
data_generator.py:181
Method__len__
(self)
data_generator.py:228
Method__len__
(self)
data_generator.py:276
Method__len__
(self)
data_generator.py:309
Method_init_weights
(self, m)
model/htsat.py:559
Functionclip_bce
Binary crossentropy loss.
utils.py:118
Functionclip_ce
(pred, target)
utils.py:125
Methodconfigure_optimizers
(self)
sed_model.py:232
Methodconv_1x1_bn
(inp, oup)
models.py:1533
Methodconv_bn
(inp, oup, stride)
models.py:1335
Methodconv_bn
(inp, oup, stride)
models.py:1520
Methodconv_dw
(inp, oup, stride)
models.py:1347
Functiondo_mixup_label
(x)
utils.py:142
Methodextra_repr
(self)
model/htsat.py:140
Methodextra_repr
(self)
model/htsat.py:263
Methodextra_repr
(self)
model/htsat.py:307
Methodextra_repr
(self)
model/htsat.py:375
Methodforward
Parameters ---------- x: input logits y: targets (multi-label binarized vector)
utils.py:38
Methodforward
(self, input, pool_size=(2, 2), pool_type='avg')
models.py:53
Methodforward
(self, input, pool_size=(2, 2), pool_type='avg')
models.py:91
Methodforward
(self, x)
models.py:126
Methodforward
Input: (batch_size, data_length)
models.py:186
Methodforward
Input: (batch_size, data_length)
models.py:273
Methodforward
Input: (batch_size, data_length)
models.py:361
Methodforward
Input: (batch_size, data_length)
models.py:444
Methodforward
Input: (batch_size, data_length)
models.py:529
Methodforward
(self, x)
models.py:614
Methodforward
(self, x)
models.py:671
Methodforward
(self, x)
models.py:763
Methodforward
Input: (batch_size, data_length)
models.py:819
Methodforward
Input: (batch_size, data_length)
models.py:906
Methodforward
Input: (batch_size, data_length)
models.py:993
Methodforward
Input: (batch_size, data_length)
models.py:1079
Methodforward
Input: (batch_size, data_length)
models.py:1170
Methodforward
Input: (batch_size, data_length)
models.py:1261
Methodforward
Input: (batch_size, data_length)
models.py:1390
Methodforward
(self, x)
models.py:1469
Methodforward
Input: (batch_size, data_length)
models.py:1571
Methodforward
(self, x, pool_size=1)
models.py:1624
Methodforward
Input: (batch_size, data_length)
models.py:1664
Methodforward
(self, x, pool_size=1)
models.py:1723
Methodforward
Input: (batch_size, data_length)
models.py:1763
Methodforward
(self, input, pool_size=1)
models.py:1855
Methodforward
Input: (batch_size, data_length)
models.py:1901
Methodforward
(self, x)
models.py:1977
← previousnext →101–200 of 248, ranked by callers