Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TypeError: umx() got an unexpected keyword argument 'target' #128

Open
jiseong1209 opened this issue Feb 1, 2023 · 0 comments
Open

TypeError: umx() got an unexpected keyword argument 'target' #128

jiseong1209 opened this issue Feb 1, 2023 · 0 comments

Comments

@jiseong1209
Copy link

jiseong1209 commented Feb 1, 2023

Hello, If you run this code:

import torch
eps = np.finfo(np.float).eps

N = track.audio.shape[0] # remember number of samples for future use
X = stft(track.audio.T, nperseg=4096, noverlap=3072)[-1]
(I, F, T) = X.shape

P = {}
model = eps

for name, source in track.sources.items():
unmix = torch.hub.load('sigsep/open-unmix-pytorch', model='umx', target=name)
P[name] = unmix(torch.tensor(track.audio.T[None, ...]).float()).detach().numpy()[:, 0, ...].transpose(1, 2, 0)
model += P[name]

The following error occurs:

TypeError Traceback (most recent call last)
in
15 for name, source in track.sources.items():
16 # compute spectrogram of target source:
---> 17 unmix = torch.hub.load('sigsep/open-unmix-pytorch', model='umx', target=name)
18 P[name] = unmix(torch.tensor(track.audio.T[None, ...]).float()).detach().numpy()[:, 0, ...].transpose(1, 2, 0)
19 # P[name] = np.abs(stft(source.audio.T, nperseg=4096, noverlap=3072)[-1])

1 frames
/usr/local/lib/python3.8/dist-packages/torch/hub.py in load(repo_or_dir, model, source, trust_repo, force_reload, verbose, skip_validation, *args, **kwargs)
540 verbose=verbose, skip_validation=skip_validation)
541
--> 542 model = _load_local(repo_or_dir, model, *args, **kwargs)
543 return model
544

/usr/local/lib/python3.8/dist-packages/torch/hub.py in _load_local(hubconf_dir, model, *args, **kwargs)
570
571 entry = _load_entry_from_hubconf(hub_module, model)
--> 572 model = entry(*args, **kwargs)
573
574 sys.path.remove(hubconf_dir)

TypeError: umx() got an unexpected keyword argument 'target'

How can we solve this problem?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant