Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZFTurbo/Music-Source-Separation-Training
/ functions
Functions
993 in github.com/ZFTurbo/Music-Source-Separation-Training
⨍
Functions
993
◇
Types & classes
292
Function
inner
(x)
models/bs_mamba2_code/attend_mamba.py:32
Function
l2norm
(t)
models/scnet/scnet_tran.py:40
Function
l2norm
(t)
models/bs_roformer/bs_roformer.py:48
Function
l2norm
(t)
models/bs_roformer/mel_band_conformer.py:47
Function
l2norm
(t)
models/bs_roformer/mel_band_roformer.py:54
Function
l2norm
(t)
models/bs_roformer/bs_conformer.py:45
Function
l2norm
(t)
models/bs_roformer/mel_band_roformer_experimental.py:48
Function
l2norm
(t)
models/bs_roformer/bs_roformer_experimental.py:41
Method
load_state_dict_in_audio
(model, pretrained_dict)
models/look2hear/models/base_model.py:48
Function
loudness_norm
( data_path: str, # output_path: str, target_lufs = -17.0, )
models/bandit/core/data/musdb/preprocess.py:214
Function
loudness_norm_one
( inputs )
models/bandit/core/data/musdb/preprocess.py:199
Method
mask
(self, x, m)
models/bandit/core/model/bsrnn/core.py:352
Method
mask
(x, m)
models/bandit/core/model/bsrnn/__init__.py:22
Function
mel_filterbank
(n_bands, fs, f_min, f_max, n_freqs)
models/bandit/core/model/bsrnn/utils.py:353
Function
mel_filterbank
(n_bands, fs, f_min, f_max, n_freqs)
models/bandit_v2/utils.py:297
Function
merge_chunks_all
( combined: torch.Tensor, original_batch_size: int, n_channel: int, n_samples:
models/bandit/core/utils/audio.py:68
Function
merge_chunks_edge
( combined: torch.Tensor, original_batch_size: int, n_channel: int, n_samples:
models/bandit/core/utils/audio.py:114
Function
minibark_filterbank
( n_bands, fs, f_min, f_max, n_freqs )
models/bandit/core/model/bsrnn/utils.py:490
Function
music_collate_fn
batch: list of elements from Dataset.__getitem__
utils/dataset.py:32
Function
musical_filterbank
(n_bands, fs, f_min, f_max, n_freqs, scale="constant")
models/bandit/core/model/bsrnn/utils.py:378
Function
musical_filterbank
(n_bands, fs, f_min, f_max, n_freqs, scale="constant")
models/bandit_v2/utils.py:325
Method
old_mask
(self, x, m)
models/bandit/core/model/bsrnn/core.py:541
Method
old_predtest_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:402
Method
on_click
(event)
gui/gui-wx.py:200
Method
on_combination_selected
(self, event)
gui/gui-wx.py:561
Method
on_download_models
(self, event)
gui/gui-wx.py:543
Method
on_enter
(event)
gui/gui-wx.py:192
Method
on_help_enter
(event)
gui/gui-wx.py:311
Method
on_help_leave
(event)
gui/gui-wx.py:315
Method
on_leave
(event)
gui/gui-wx.py:196
Method
on_link_click
(self, event)
gui/gui-wx.py:600
Method
on_page_load
(self, event)
gui/gui-wx.py:606
Method
on_paint
(self, event)
gui/gui-wx.py:105
Method
on_save_combination
(self, event)
gui/gui-wx.py:546
Method
on_test_epoch_end
(self)
models/bandit/core/__init__.py:503
Method
on_test_epoch_start
(self)
models/bandit/core/__init__.py:477
Method
on_toggle
(self, event)
gui/gui-wx.py:74
Method
on_train_batch_end
( self, outputs: STEP_OUTPUT, batch: BatchedDataDict, batch_idx: int )
models/bandit/core/__init__.py:355
Method
on_validation_epoch_end
(self)
models/bandit/core/__init__.py:390
Function
pack_one
(t, pattern)
models/scnet/scnet_tran.py:26
Function
pad_at_dim
(t, pad, dim=-1, value=0.)
models/scnet/scnet_tran.py:34
Function
pad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_conformer.py:41
Function
pad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_roformer.py:48
Function
pad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/bs_conformer.py:39
Function
pad_at_dim
(t, pad, dim=-1, value=0.)
models/bs_roformer/mel_band_roformer_experimental.py:42
Method
pad_input
Zero-padding input according to window/stride size.
models/ts_bs_mamba2.py:216
Function
pad_to_appropriate_length
(x, lcm)
models/look2hear/models/base_model.py:21
Method
predict_step
( self, batch: BatchedDataDict, batch_idx: int = 0, dataloader
models/bandit/core/__init__.py:514
Function
prepare_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
Function
preprocess
( data_path: str, output_path: str, fs: int )
models/bandit/core/data/dnr/preprocess.py:33
Function
preprocess
( analysis_stem: str, output_path: str = "/data/MUSDB18/HQ/saded-np", fs: int = 44100,
models/bandit/core/data/musdb/preprocess.py:159
Function
process_chunk_worker
(args)
utils/dataset.py:215
Function
process_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
Function
process_one
(inputs: Tuple[str, str, int])
models/bandit/core/data/dnr/preprocess.py:10
Function
register_model
Register a custom model, gettable with `models.get`. Args: custom_model: Custom model to register.
models/look2hear/models/__init__.py:17
Method
run_inference
(self, event)
gui/gui-wx.py:452
Function
run_subprocess
(cmd, output_queue)
gui/gui-wx.py:14
Method
run_training
(self, event)
gui/gui-wx.py:403
Method
sample_rate
Sample rate of the audio that the model is trained on. :type: float
models/bandit/core/metrics/_squim.py:362
Method
sample_rate
(self,)
models/look2hear/models/base_model.py:44
Method
save_to_file
(self, dir_path: str)
models/bandit/core/model/bsrnn/utils.py:337
Method
save_to_file
(self, dir_path: str)
models/bandit_v2/utils.py:281
Method
separate
(self, batch)
models/bandit_v2/bandit.py:351
Method
serialize
(self)
models/look2hear/models/base_model.py:78
Method
set_predict_output_path
(self, path: str)
models/bandit/core/__init__.py:719
Method
step
(self, closure: Closure = None)
utils/muon.py:345
Method
step
(self, closure: Closure = None)
utils/muon.py:546
Method
step
(self, closure: Closure = None)
utils/muon.py:744
Function
test_export_jit_script
(net, x)
models/ex_bi_mamba2.py:266
Function
test_export_onnx
(net, x)
models/ex_bi_mamba2.py:275
Method
test_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:480
Function
train_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
Method
training_step
(self, batch: BatchedDataDict)
models/bandit/core/__init__.py:336
Function
triangular_bark_filterbank
( n_bands, fs, f_min, f_max, n_freqs )
models/bandit/core/model/bsrnn/utils.py:452
Function
unpack_one
(t, ps, pattern)
models/scnet/scnet_tran.py:30
Method
update
(self, preds: torch.Tensor, target: torch.Tensor)
models/bandit/core/metrics/snr.py:50
Method
valid_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
Function
validate_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
Method
validation_step
( self, batch: BatchedDataDict, batch_idx: int, dataloader_idx
models/bandit/core/__init__.py:363
Method
version1
(self)
models/bandit/core/model/bsrnn/utils.py:106
Method
version1
(self)
models/bandit_v2/utils.py:100
Method
version2
(self)
models/bandit/core/model/bsrnn/utils.py:111
Method
version2
(self)
models/bandit_v2/utils.py:105
Method
version3
(self)
models/bandit/core/model/bsrnn/utils.py:123
Method
version3
(self)
models/bandit_v2/utils.py:115
Method
version4
(self)
models/bandit/core/model/bsrnn/utils.py:135
Method
version4
(self)
models/bandit_v2/utils.py:125
Method
version5
(self)
models/bandit/core/model/bsrnn/utils.py:152
Method
version5
(self)
models/bandit_v2/utils.py:138
Method
version6
(self)
models/bandit/core/model/bsrnn/utils.py:168
Method
version6
(self)
models/bandit_v2/utils.py:150
Method
version7
(self)
models/bandit/core/model/bsrnn/utils.py:189
Method
version7
(self)
models/bandit_v2/utils.py:165
← previous
901–993 of 993, ranked by callers