Skip to content

Commit

Permalink
add logging
Browse files Browse the repository at this point in the history
  • Loading branch information
priyakasimbeg committed Sep 28, 2023
1 parent 784230f commit e83937b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions submission_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -531,6 +531,7 @@ def score_submission_on_workload(workload: spec.Workload,
if 'imagenet' not in workload_name:
imagenet_v2_data_dir = None
logging.info(f"Workload step hint: {workload.step_hint}")
logging.info(f'Workload setting step hint: {target_setting_step_hint}')
timing, metrics = train_once(workload, global_batch_size,
global_eval_batch_size,
data_dir, imagenet_v2_data_dir,
Expand Down

0 comments on commit e83937b

Please sign in to comment.