1 files changed, 3 insertions, 3 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index b029eab..4399a60 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -141,6 +141,7 @@ class Pix2PixHDTrain extends Component {
               title="Tag this epoch"
               value={this.state.augment_name}
               onChange={this.handleChange}
+              className='small'
             />
             <Button
               title="Augment dataset"
@@ -171,10 +172,9 @@ class Pix2PixHDTrain extends Component {
               value={this.props.runner.gpu.status === 'IDLE' ? "Idle" : "Interrupt"}
               onClick={() => this.interrupt()}
             />
+            <CurrentTask />
           </Group>
-        </div>
-        <div>
-          <CurrentTask />
+
           {!!queue.queue.length &&
             <Group title='Upcoming Tasks'>
               <TaskList tasks={queue.queue.map(id => queue.tasks[id])} sort="date asc" />