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
Methodforward
(self, x)
models.py:2067
Methodforward
Input: (batch_size, data_length)
models.py:2109
Methodforward
Input: (batch_size, data_length)
models.py:2164
Methodforward
(self, input, pool_size)
models.py:2217
Methodforward
Input: (batch_size, data_length)
models.py:2272
Methodforward
Input: (batch_size, data_length)
models.py:2370
Methodforward
Input: (batch_size, data_length)
models.py:2484
Methodforward
Input: (batch_size, data_length)
models.py:2595
Methodforward
Input: (batch_size, data_length)
models.py:2692
Methodforward
Input: (batch_size, data_length)
models.py:2783
Methodforward
Input: (batch_size, data_length)
models.py:2876
Methodforward
Input: (batch_size, data_length)
models.py:2967
Methodforward
Input: (batch_size, data_length)
models.py:3059
Methodforward
Input: (batch_size, data_length)
models.py:3158
Methodforward
Input: (batch_size, data_length)
models.py:3262
Methodforward
(self, x, mix_lambda = None)
sed_model.py:53
Methodforward
(self, x, sed_index, mix_lambda = None)
sed_model.py:277
Methodforward
(self, x)
model/layers.py:59
Methodforward
(self, x)
model/layers.py:84
Methodforward
(self, x)
model/layers.py:106
Methodforward
Args: x: input features with shape of (num_windows*B, N, C) mask: (0/-inf) mask with shape of (num_windows, Wh*Ww, Wh
model/htsat.py:107
Methodforward
(self, x)
model/htsat.py:220
Methodforward
x: B, H*W, C
model/htsat.py:284
Methodforward
(self, x)
model/htsat.py:359
Methodforward
(self, x: torch.Tensor, mixup_lambda = None, infer_mode = False)
model/htsat.py:762
Methodinference
(self, x)
sed_model.py:57
Functioninit_hier_head
(class_map, num_class)
utils.py:214
Functionlabel_enhance
(preds)
fl_evaluate.py:38
Functionlecun_normal_
(tensor)
model/layers.py:194
Methodlr_foo
(epoch)
sed_model.py:240
Methodno_weight_decay
(self)
model/htsat.py:569
Methodno_weight_decay_keywords
(self)
model/htsat.py:573
Functionparse
(x)
model/layers.py:21
Methodpredict
Run a single prediction on the model
predict.py:48
Functionprepprocess_audio
(track, ofs, rfs, mono_type = "mix")
utils.py:197
Functionscratch_optim
(preds, threshold, sec_threshold, mode)
fl_evaluate.py:21
Methodsetup
Load the model into memory to make running multiple predictions efficient
predict.py:15
Methodtest_dataloader
(self)
main.py:68
Methodtest_epoch_end
(self, test_step_outputs)
sed_model.py:179
Methodtest_epoch_end
(self, test_step_outputs)
sed_model.py:331
Methodtest_step
(self, batch, batch_idx)
sed_model.py:155
Methodtest_step
(self, batch, batch_idx)
sed_model.py:303
Methodtrain_dataloader
(self)
main.py:48
Methodtraining_epoch_end
(self, outputs)
sed_model.py:83
Methodtraining_step
(self, batch, batch_idx)
sed_model.py:66
Methodval_dataloader
(self)
main.py:58
Methodvalidation_epoch_end
(self, validation_step_outputs)
sed_model.py:96
Methodvalidation_step
(self, batch, batch_idx)
sed_model.py:92
← previous201–248 of 248, ranked by callers