File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -352,7 +352,6 @@ def train_once(
352352 log_dir , flags .FLAGS , hyperparameters
353353 )
354354 workload .attach_metrics_logger (metrics_logger )
355- step_10_end_time = None
356355 global_start_time = get_time ()
357356 train_state ['last_step_end_time' ] = global_start_time
358357
@@ -409,21 +408,6 @@ def train_once(
409408 train_state ['training_complete' ] = True
410409
411410 train_step_end_time = get_time ()
412- if global_step == 11 :
413- step_10_end_time = train_step_end_time
414-
415- # Log step time every 100 steps
416- if (global_step - 1 ) % 100 == 0 and workload .metrics_logger is not None :
417- if step_10_end_time is not None and global_step > 11 :
418- elapsed_time_ms = (train_step_end_time - step_10_end_time ) * 1000.0
419- elapsed_steps = global_step - 11
420- avg_step_time_ms = elapsed_time_ms / elapsed_steps
421- else :
422- avg_step_time_ms = 0.0
423- workload .metrics_logger .append_scalar_metrics (
424- {'step_time_ms' : avg_step_time_ms },
425- global_step - 1 ,
426- )
427411
428412 train_state ['accumulated_submission_time' ] += (
429413 train_step_end_time - train_state ['last_step_end_time' ]
You can’t perform that action at this time.
0 commit comments