Wave-U-Net-Pytorch

Wave-U-Net-Pytorch copied to clipboard

fixed #14 tested as well on depth 2 and 3 using train.py

B-lanc

Changing the depth to anything other than 1 will result in this error > RuntimeError: Given groups=1, weight of size [512, 512, 5], expected input[1, 1024, 357] to have 512...

B-lanc

Update Replicate to use multiple outputs

2

Hi @f90! This is a follow-up to #9, where we added the Cog configuration and a [Replicate demo](https://replicate.com/f90/wave-u-net-pytorch). We've now added support for multiple outputs, so we can display each...

bfirsh

Difference output layer doesn't exist in pytorch version.

1

Doesn't difference output layer exist in pytorch version? In tensorflow version I could find 'output_type' parameter, but not in pytorch version. Is there any reason for not including that layer?

jc5201

Cannot train a model with the same accuracy as the paper's

1

Thanks for such excellent work! The code is great and well-organized. But I met this problem during training. I tried evaluating the SDR for both pretrained model and the model...

sewichou

Traceback (most recent call last): File "F:/1Python/1_code_xunlian_from_githubOrOther/2_wave_u_net/Wave-U-Net-Pytorch-fuben/train.py", line 243, in main(args) File "F:/1Python/1_code_xunlian_from_githubOrOther/2_wave_u_net/Wave-U-Net-Pytorch-fuben/train.py", line 100, in main utils.set_cyclic_lr(optimizer, example_num, len(train_data) // args.batch_size, args.cycles, args.min_lr, args.lr) File "F:\1Python\1_code_xunlian_from_githubOrOther\2_wave_u_net\Wave-U-Net-Pytorch-fuben\utils.py", line 16, in...

xiao-c-y

reinerterig

Support for Apple Metal (MPS) backend

1

Please add support for the MPS backend as you do for cuda: ```python if torch.backends.mps.is_available(): mps = torch.device("mps") model = model_utils.DataParallel(model) model.to(mps) # ... and so on... x = x.to(mps)...

Archie3d

Maybe I missed sth.

1

FileNotFoundError: [Errno 2] No such file or directory: 'E:\\xxx\\Dataset\\Dataset\\musdb18\\train\\A Classic Education - NightOwl.stem.mp4\\bass.wav' How can I solve it T_T

Piscis-Austrini

Bumps [torch](https://github.com/pytorch/pytorch) from 1.4.0 to 2.2.0. Release notes Sourced from torch's releases. PyTorch 2.2: FlashAttention-v2, AOTInductor PyTorch 2.2 Release Notes Highlights Backwards Incompatible Changes Deprecations New Features Improvements Bug fixes...

dependabot[bot]