Skip to content

Add PyTorch converted model#4

Open
jdasam wants to merge 7 commits intokeums:masterfrom
jdasam:master
Open

Add PyTorch converted model#4
jdasam wants to merge 7 commits intokeums:masterfrom
jdasam:master

Conversation

@jdasam
Copy link

@jdasam jdasam commented Jan 18, 2021

I've converted the pre-trained model to PyTorch, so that users can run the code with torch library instead of keras.

The followings are updated:

  • Melody ResNet model in PyTorch
  • Converted parameters of pre-trained Keras weights to PyTorch weights
  • Jupyter notebook for weight conversion
  • An option (-torch, --run_on_torch) for melodyExtraction_NS.py to run inference with torch model
    • add an explanation on the readme.md

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

Successfully merging this pull request may close these issues.

1 participant