Skip to content

Commit

Permalink
correct resnet estimate time (tinygrad#4169)
Browse files Browse the repository at this point in the history
7.99 hours was rendered as 7h0m.
  • Loading branch information
chenyuxyz committed Apr 14, 2024
1 parent ea18d28 commit e20d6f9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/mlperf/model_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -170,8 +170,8 @@ def data_get(it):

if i == BENCHMARK:
median_step_time = sorted(step_times)[(BENCHMARK + 1) // 2] # in seconds
estimated_total_hours = median_step_time * steps_in_train_epoch * epochs / 60 / 60
print(f"Estimated training time: {estimated_total_hours:.0f}h{(estimated_total_hours - int(estimated_total_hours)) * 60:.0f}m")
estimated_total_minutes = int(median_step_time * steps_in_train_epoch * epochs / 60)
print(f"Estimated training time: {estimated_total_minutes // 60}h{estimated_total_minutes % 60}m")
# if we are doing beam search, run the first eval too
if BEAM.value and e == start_epoch: break
return
Expand Down

0 comments on commit e20d6f9

Please sign in to comment.