backend uses attention_rnn now

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

BIN
server/attention_rnn.mag Normal file

Binary file not shown.

View File

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