ale/aiexperiments-ai-duet
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

backend uses attention_rnn now

Browse Source 
i found it was producing better results, but providing both models
makes it easy to switch back and forth
This commit is contained in:
Yotam Mann 2017-02-13 17:29:42 -05:00
parent 39d9a2ae72
commit 06e45fa266
2 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
server/attention_rnn.mag Normal file
View File

Binary file not shown.

2
server/predict.py
View File

@ -29,7 +29,7 @@ import time
import tempfile
import pretty_midi
BUNDLE_NAME = 'basic_rnn'
BUNDLE_NAME = 'attention_rnn'
config = magenta.models.melody_rnn.melody_rnn_model.default_configs[BUNDLE_NAME]
bundle_file = magenta.music.read_bundle_file(os.path.abspath(BUNDLE_NAME+'.mag'))