Skip to content

Commit 6e2bad4

Browse files
committed
remove logging from submission runner
1 parent 400640a commit 6e2bad4

1 file changed

Lines changed: 0 additions & 16 deletions

File tree

submission_runner.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff 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']

0 commit comments

Comments
 (0)