diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-08-12 00:43:39 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-08-12 00:43:39 +0200 |
| commit | f3acc04e3a7c1248531df8397277c41a4964feeb (patch) | |
| tree | 5565f97c3646277f56a4d10a893ac8b092102a2e /app | |
| parent | 9c1f011fb187cd7ae77bc79a429b238c8b5fc570 (diff) | |
update samplernn bin
Diffstat (limited to 'app')
| -rw-r--r-- | app/relay/interpreters.js | 5 | ||||
| -rw-r--r-- | app/relay/modules/samplernn.js | 4 |
2 files changed, 7 insertions, 2 deletions
diff --git a/app/relay/interpreters.js b/app/relay/interpreters.js index b1507e6..d892acf 100644 --- a/app/relay/interpreters.js +++ b/app/relay/interpreters.js @@ -18,6 +18,11 @@ export default { params: ['-u'], gpu: false, }, + pytorch_samplernn: { + cmd: process.env.PYTORCH_SAMPLERNN_BIN, + params: ['-u'], + gpu: true, + }, pytorch: { cmd: process.env.PYTORCH_BIN, params: ['-u'], diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js index 900679d..d25698f 100644 --- a/app/relay/modules/samplernn.js +++ b/app/relay/modules/samplernn.js @@ -23,7 +23,7 @@ const fetch = { } } const train = { - type: 'pytorch', + type: 'pytorch_samplernn', script: 'train.py', params: (task) => { return [ @@ -52,7 +52,7 @@ const train = { after: 'publish', } const generate = { - type: 'pytorch', + type: 'pytorch_samplernn', script: 'generate.py', params: (task) => { return [ |