@@ -105,15 +105,15 @@ def on_step_end(self, status):
105
105
epoch_id = status ['epoch_id' ]
106
106
step_id = status ['step_id' ]
107
107
steps_per_epoch = status ['steps_per_epoch' ]
108
- training_staus = status ['training_staus ' ]
108
+ training_status = status ['training_status ' ]
109
109
batch_time = status ['batch_time' ]
110
110
data_time = status ['data_time' ]
111
111
112
112
epoches = self .model .cfg .epoch
113
113
batch_size = self .model .cfg ['{}Reader' .format (mode .capitalize (
114
114
))]['batch_size' ]
115
115
116
- logs = training_staus .log ()
116
+ logs = training_status .log ()
117
117
space_fmt = ':' + str (len (str (steps_per_epoch ))) + 'd'
118
118
if step_id % self .model .cfg .log_iter == 0 :
119
119
eta_steps = (epoches - epoch_id ) * steps_per_epoch - step_id
@@ -278,8 +278,8 @@ def on_step_end(self, status):
278
278
mode = status ['mode' ]
279
279
if dist .get_world_size () < 2 or dist .get_rank () == 0 :
280
280
if mode == 'train' :
281
- training_staus = status ['training_staus ' ]
282
- for loss_name , loss_value in training_staus .get ().items ():
281
+ training_status = status ['training_status ' ]
282
+ for loss_name , loss_value in training_status .get ().items ():
283
283
self .vdl_writer .add_scalar (loss_name , loss_value ,
284
284
self .vdl_loss_step )
285
285
self .vdl_loss_step += 1
@@ -401,7 +401,7 @@ def on_step_end(self, status):
401
401
mode = status ['mode' ]
402
402
if dist .get_world_size () < 2 or dist .get_rank () == 0 :
403
403
if mode == 'train' :
404
- training_status = status ['training_staus ' ].get ()
404
+ training_status = status ['training_status ' ].get ()
405
405
for k , v in training_status .items ():
406
406
training_status [k ] = float (v )
407
407
0 commit comments