diff options
Diffstat (limited to 'app/client/modules/pix2pixhd/views/pix2pixhd.train.js')
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index ace93e4..b029eab 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -64,6 +64,10 @@ class Pix2PixHDTrain extends Component { console.log('name', name, 'value', value) this.setState({ [name]: value }) } + interrupt(){ + this.props.actions.queue.stop_task('gpu') + } + render(){ if (this.props.pix2pixhd.loading) { return <Loading progress={this.props.pix2pixhd.progress} /> @@ -89,9 +93,9 @@ class Pix2PixHDTrain extends Component { } }).filter(n => !!n && !!n.options.length).sort((a,b) => a.name.localeCompare(b.name)) - console.log('state', this.props.pix2pixhd.data.epochs) + // console.log('state', this.props.pix2pixhd.data.epochs) // console.log(this.state.checkpoint_name, this.state.epoch) - + console.log(queue) return ( <div className='app pix2pixhd'> <div class='heading'> @@ -160,12 +164,20 @@ class Pix2PixHDTrain extends Component { onClick={() => this.props.remote.clear_recursive_task(this.state.checkpoint_name)} /> </Group> + + <Group title='Status'> + <Button + title="GPU" + value={this.props.runner.gpu.status === 'IDLE' ? "Idle" : "Interrupt"} + onClick={() => this.interrupt()} + /> + </Group> </div> <div> <CurrentTask /> {!!queue.queue.length && <Group title='Upcoming Tasks'> - <TaskList tasks={queue.queue.map(id => queue.tasks[id])} /> + <TaskList tasks={queue.queue.map(id => queue.tasks[id])} sort="date asc" /> </Group> } </div> @@ -176,6 +188,7 @@ class Pix2PixHDTrain extends Component { const mapStateToProps = state => ({ pix2pixhd: state.module.pix2pixhd, + runner: state.system.runner, queue: state.queue, }) |