diff --git a/mindinsight/ui/src/views/train-manage/data-traceback.vue b/mindinsight/ui/src/views/train-manage/data-traceback.vue
index 5db62e2dfc63863d2f36dcba1c3a5392d2d8612d..2bf0f4487116e62beb63a66df945e485283adf1d 100644
--- a/mindinsight/ui/src/views/train-manage/data-traceback.vue
+++ b/mindinsight/ui/src/views/train-manage/data-traceback.vue
@@ -416,7 +416,7 @@ export default {
         this.$t('modelTraceback.testSampleNum'),
         this.$t('modelTraceback.learningRate'),
         'epoch',
-        'steps',
+        'steps_per_epoch',
         this.$t('modelTraceback.deviceNum'),
         this.$t('modelTraceback.modelSize'),
         this.$t('modelTraceback.lossFunc'),
@@ -465,7 +465,7 @@ export default {
             label: 'epoch',
           },
           batch_size: {
-            label: 'steps',
+            label: 'steps_per_epoch',
           },
           device_num: {
             label: this.$t('modelTraceback.deviceNum'),
diff --git a/mindinsight/ui/src/views/train-manage/model-traceback.vue b/mindinsight/ui/src/views/train-manage/model-traceback.vue
index 6a564dd18eb2b56a43fe85753822bc75e4ddaa44..d9c92d4702790edf5204719da8c7fbb97e988e52 100644
--- a/mindinsight/ui/src/views/train-manage/model-traceback.vue
+++ b/mindinsight/ui/src/views/train-manage/model-traceback.vue
@@ -176,7 +176,7 @@ limitations under the License.
                              :prop="key"
                              :label="table.columnOptions[key].label"
                              show-overflow-tooltip
-                             min-width="150"
+                             min-width="154"
                              sortable="custom">
               <template slot="header"
                         slot-scope="scope">
@@ -810,7 +810,7 @@ export default {
             required: false,
           },
           batch_size: {
-            label: 'steps',
+            label: 'steps_per_epoch',
             required: false,
           },
           device_num: {