recipes.cad2.task2.ConvTasNet package

Subpackages

• recipes.cad2.task2.ConvTasNet.local package
  • Submodules
  • recipes.cad2.task2.ConvTasNet.local.cad2task2_dataloader module
    • Compose
    • RebalanceMusicDataset
      • RebalanceMusicDataset.accompaniment_tracks
      • RebalanceMusicDataset.dataset_name
      • RebalanceMusicDataset.get_infos()
      • RebalanceMusicDataset.repeated_instruments
      • RebalanceMusicDataset.root_path
    • augment_channelswap()
    • augment_gain()
    • get_audio_durations()
  • recipes.cad2.task2.ConvTasNet.local.tasnet module
    • ChannelwiseLayerNorm
      • ChannelwiseLayerNorm.forward()
      • ChannelwiseLayerNorm.reset_parameters()
    • Chomp1d
      • Chomp1d.forward()
    • ConvTasNetStereo
      • ConvTasNetStereo.forward()
      • ConvTasNetStereo.get_model_args()
      • ConvTasNetStereo.get_state_dict()
      • ConvTasNetStereo.serialize()
      • ConvTasNetStereo.valid_length()
    • Decoder
      • Decoder.forward()
    • DepthwiseSeparableConv
      • DepthwiseSeparableConv.forward()
    • Encoder
      • Encoder.forward()
    • GlobalLayerNorm
      • GlobalLayerNorm.forward()
      • GlobalLayerNorm.reset_parameters()
    • TemporalBlock
      • TemporalBlock.forward()
    • TemporalConvNet
      • TemporalConvNet.forward()
    • chose_norm()
    • overlap_and_add()
  • Module contents
    • Compose
    • ConvTasNetStereo
      • ConvTasNetStereo.forward()
      • ConvTasNetStereo.get_model_args()
      • ConvTasNetStereo.get_state_dict()
      • ConvTasNetStereo.serialize()
      • ConvTasNetStereo.valid_length()
    • RebalanceMusicDataset
      • RebalanceMusicDataset.accompaniment_tracks
      • RebalanceMusicDataset.dataset_name
      • RebalanceMusicDataset.get_infos()
      • RebalanceMusicDataset.repeated_instruments
      • RebalanceMusicDataset.root_path
    • augment_channelswap()
    • augment_gain()
    • overlap_and_add()

Submodules

recipes.cad2.task2.ConvTasNet.eval module

Model evaluation script for Conv-TasNet.

recipes.cad2.task2.ConvTasNet.eval.eval_track(reference, user_estimates, eval_targets=None, track_name='', mode='v4', win=1.0, hop=1.0, sample_rate=44100)

recipes.cad2.task2.ConvTasNet.eval.main(conf)

recipes.cad2.task2.ConvTasNet.train module

recipes.cad2.task2.ConvTasNet.train.main(conf)

Module contents