diff options
Diffstat (limited to 'app/client/modules/samplernn/samplernn.datasets.js')
| -rw-r--r-- | app/client/modules/samplernn/samplernn.datasets.js | 21 |
1 files changed, 9 insertions, 12 deletions
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js index d29cf6a..b4f0c12 100644 --- a/app/client/modules/samplernn/samplernn.datasets.js +++ b/app/client/modules/samplernn/samplernn.datasets.js @@ -72,9 +72,9 @@ class SampleRNNDatasets extends Component { if (!folder || !folder.name) return return ( <div class='rows params datasets'> - <div class='row dataset'> + <div class='row row-heading dataset'> <div class='col'>input</div> - <div class='col'></div> + <div class='col'>status</div> <div class='col'>checkpoint</div> <div class='col'>output</div> </div> @@ -83,11 +83,14 @@ class SampleRNNDatasets extends Component { ) } renderGroups(){ - const { samplernn, onPickDataset, onPickFile, actions } = this.props + const { samplernn, runner, onPickDataset, onPickFile, datasetActions } = this.props const folder = samplernn.folder const { mapFn, sortFn } = util.sort.orderByFn('date desc') + const moduleOnGPU = runner.gpu.task.module === samplernnModule.name const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => { const dataset = pair[1] + const isProcessing = moduleOnGPU && runner.gpu.task.dataset === dataset + const status = isProcessing ? util.gerund(runner.gpu.task.activity) : '' return ( <div className='row dataset' onClick={() => onPickDataset && onPickDataset(dataset)}> {this.props.beforeRow && this.props.beforeRow(dataset)} @@ -103,15 +106,9 @@ class SampleRNNDatasets extends Component { /> } </div> - <div className='col quiet hiddenActions'> - <div class='subtext'>{dataset.isBuilt ? 'cached' : ''}</div> - <div class='actions'> - <span class='link' onClick={() => actions.train_task_now(dataset, 1)}>train</span> - <span class='link' onClick={() => actions.train_task_now(dataset, 2)}>2x</span> - <span class='link' onClick={() => actions.train_task_now(dataset, 4)}>4x</span> - <span class='link' onClick={() => actions.train_task_now(dataset, 6)}>6x</span> - <span class='link' onClick={() => actions.train_task_now(dataset, 18)}>18x</span> - </div> + <div className={'col quiet ' + (dataset.isBuilt ? 'built' : 'not_built')}> + {status} + {this.props.datasetActions && this.props.datasetActions(dataset)} </div> <div className='col checkpoint'> {!!dataset.checkpoints.length && |