diff options
Diffstat (limited to 'app/relay/modules/samplernn.js')
| -rw-r--r-- | app/relay/modules/samplernn.js | 4 |
1 files changed, 2 insertions, 2 deletions
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 [ |