MCPcopy Create free account

hub / github.com/ZFTurbo/Music-Source-Separation-Training / functions

Functions993 in github.com/ZFTurbo/Music-Source-Separation-Training

Functioninner
(x)
models/bs_mamba2_code/attend_mamba.py:32
Functionl2norm
(t)
models/scnet/scnet_tran.py:40
Functionl2norm
(t)
models/bs_roformer/bs_roformer.py:48
Functionl2norm
(t)
models/bs_roformer/mel_band_conformer.py:47
Functionl2norm
(t)
models/bs_roformer/mel_band_roformer.py:54
Functionl2norm
(t)
models/bs_roformer/bs_conformer.py:45
Functionl2norm
(t)
models/bs_roformer/mel_band_roformer_experimental.py:48
Functionl2norm
(t)
models/bs_roformer/bs_roformer_experimental.py:41
Methodload_state_dict_in_audio
(model, pretrained_dict)
models/look2hear/models/base_model.py:48
Functionloudness_norm
( data_path: str, # output_path: str, target_lufs = -17.0, )
models/bandit/core/data/musdb/preprocess.py:214
Functionloudness_norm_one
( inputs )
models/bandit/core/data/musdb/preprocess.py:199
Methodmask
(self, x, m)
models/bandit/core/model/bsrnn/core.py:352
Methodmask
(x, m)
models/bandit/core/model/bsrnn/__init__.py:22
Functionmel_filterbank
(n_bands, fs, f_min, f_max, n_freqs)
models/bandit/core/model/bsrnn/utils.py:353
Functionmel_filterbank
(n_bands, fs, f_min, f_max, n_freqs)
models/bandit_v2/utils.py:297
Functionmerge_chunks_all
( combined: torch.Tensor, original_batch_size: int, n_channel: int, n_samples:
models/bandit/core/utils/audio.py:68
Functionmerge_chunks_edge
( combined: torch.Tensor, original_batch_size: int, n_channel: int, n_samples:
models/bandit/core/utils/audio.py:114
Functionminibark_filterbank
( n_bands, fs, f_min, f_max, n_freqs )
models/bandit/core/model/bsrnn/utils.py:490
Functionmusic_collate_fn
batch: list of elements from Dataset.__getitem__
utils/dataset.py:32
Functionmusical_filterbank
(n_bands, fs, f_min, f_max, n_freqs, scale="constant")
models/bandit/core/model/bsrnn/utils.py:378
Functionmusical_filterbank
(n_bands, fs, f_min, f_max, n_freqs, scale="constant")
models/bandit_v2/utils.py:325
Methodold_mask
(self, x, m)
models/bandit/core/model/bsrnn/core.py:541
Methodold_predtest_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:402
Methodon_click
(event)
gui/gui-wx.py:200
Methodon_combination_selected
(self, event)
gui/gui-wx.py:561
Methodon_download_models
(self, event)
gui/gui-wx.py:543
Methodon_enter
(event)
gui/gui-wx.py:192
Methodon_help_enter
(event)
gui/gui-wx.py:311
Methodon_help_leave
(event)
gui/gui-wx.py:315
Methodon_leave
(event)
gui/gui-wx.py:196
Methodon_link_click
(self, event)
gui/gui-wx.py:600
Methodon_page_load
(self, event)
gui/gui-wx.py:606
Methodon_paint
(self, event)
gui/gui-wx.py:105
Methodon_save_combination
(self, event)
gui/gui-wx.py:546
Methodon_test_epoch_end
(self)
models/bandit/core/__init__.py:503
Methodon_test_epoch_start
(self)
models/bandit/core/__init__.py:477
Methodon_toggle
(self, event)
gui/gui-wx.py:74
Methodon_train_batch_end
( self, outputs: STEP_OUTPUT, batch: BatchedDataDict, batch_idx: int )
models/bandit/core/__init__.py:355
Methodon_validation_epoch_end
(self)
models/bandit/core/__init__.py:390
Functionpack_one
(t, pattern)
models/scnet/scnet_tran.py:26
Functionpad_at_dim
(t, pad, dim=-1, value=0.)
models/scnet/scnet_tran.py:34
Functionpad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_conformer.py:41
Functionpad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_roformer.py:48
Functionpad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/bs_conformer.py:39
Functionpad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_roformer_experimental.py:42
Methodpad_input
Zero-padding input according to window/stride size.
models/ts_bs_mamba2.py:216
Functionpad_to_appropriate_length
(x, lcm)
models/look2hear/models/base_model.py:21
Methodpredict_step
( self, batch: BatchedDataDict, batch_idx: int = 0, dataloader
models/bandit/core/__init__.py:514
Functionprepare_muon_parameters
Prepare the parameters for Muon optimizer. Be careful at using this function to prepare the parameters for Muon optimizer. It's not likely acting
utils/muon.py:815
Functionpreprocess
( data_path: str, output_path: str, fs: int )
models/bandit/core/data/dnr/preprocess.py:33
Functionpreprocess
( analysis_stem: str, output_path: str = "/data/MUSDB18/HQ/saded-np", fs: int = 44100,
models/bandit/core/data/musdb/preprocess.py:159
Functionprocess_chunk_worker
(args)
utils/dataset.py:215
Functionprocess_folder_wrapper
A wrapper function for 'process_folder' that unpacks the arguments. Parameters: - args (Tuple[str, str, List[str], bool]): A Tuple contai
scripts/moises_to_musdb.py:174
Functionprocess_one
(inputs: Tuple[str, str, int])
models/bandit/core/data/dnr/preprocess.py:10
Functionregister_model
Register a custom model, gettable with `models.get`. Args: custom_model: Custom model to register.
models/look2hear/models/__init__.py:17
Methodrun_inference
(self, event)
gui/gui-wx.py:452
Functionrun_subprocess
(cmd, output_queue)
gui/gui-wx.py:14
Methodrun_training
(self, event)
gui/gui-wx.py:403
Methodsample_rate
Sample rate of the audio that the model is trained on. :type: float
models/bandit/core/metrics/_squim.py:362
Methodsample_rate
(self,)
models/look2hear/models/base_model.py:44
Methodsave_to_file
(self, dir_path: str)
models/bandit/core/model/bsrnn/utils.py:337
Methodsave_to_file
(self, dir_path: str)
models/bandit_v2/utils.py:281
Methodseparate
(self, batch)
models/bandit_v2/bandit.py:351
Methodserialize
(self)
models/look2hear/models/base_model.py:78
Methodset_predict_output_path
(self, path: str)
models/bandit/core/__init__.py:719
Methodstep
(self, closure: Closure = None)
utils/muon.py:345
Methodstep
(self, closure: Closure = None)
utils/muon.py:546
Methodstep
(self, closure: Closure = None)
utils/muon.py:744
Functiontest_export_jit_script
(net, x)
models/ex_bi_mamba2.py:266
Functiontest_export_onnx
(net, x)
models/ex_bi_mamba2.py:275
Methodtest_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:480
Functiontrain_model_single
Trains the model based on the provided arguments, including data preparation, optimizer setup, and loss calculation. The model is trained for
train_ddp.py:14
Methodtraining_step
(self, batch: BatchedDataDict)
models/bandit/core/__init__.py:336
Functiontriangular_bark_filterbank
( n_bands, fs, f_min, f_max, n_freqs )
models/bandit/core/model/bsrnn/utils.py:452
Functionunpack_one
(t, ps, pattern)
models/scnet/scnet_tran.py:30
Methodupdate
(self, preds: torch.Tensor, target: torch.Tensor)
models/bandit/core/metrics/snr.py:50
Methodvalid_length
Return a length that is appropriate for evaluation. In our case, always return the training length, unless it is smaller than
models/demucs4ht.py:518
Functionvalidate_in_subprocess
Perform validation on a subprocess with multi-processing support. Each process handles inference on a subset of the mixture files and updat
valid.py:523
Methodvalidation_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:363
Methodversion1
(self)
models/bandit/core/model/bsrnn/utils.py:106
Methodversion1
(self)
models/bandit_v2/utils.py:100
Methodversion2
(self)
models/bandit/core/model/bsrnn/utils.py:111
Methodversion2
(self)
models/bandit_v2/utils.py:105
Methodversion3
(self)
models/bandit/core/model/bsrnn/utils.py:123
Methodversion3
(self)
models/bandit_v2/utils.py:115
Methodversion4
(self)
models/bandit/core/model/bsrnn/utils.py:135
Methodversion4
(self)
models/bandit_v2/utils.py:125
Methodversion5
(self)
models/bandit/core/model/bsrnn/utils.py:152
Methodversion5
(self)
models/bandit_v2/utils.py:138
Methodversion6
(self)
models/bandit/core/model/bsrnn/utils.py:168
Methodversion6
(self)
models/bandit_v2/utils.py:150
Methodversion7
(self)
models/bandit/core/model/bsrnn/utils.py:189
Methodversion7
(self)
models/bandit_v2/utils.py:165
← previous901–993 of 993, ranked by callers