Skip to content

Commit 422a494

Browse files
authored
Fix type error in performance_profiling/pathology/train_evaluate_nvtx.py (#1416)
Fixes #1415 . ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Avoid including large-size files in the PR. - [ ] Clean up long text outputs from code cells in the notebook. - [ ] For security purposes, please check the contents and remove any sensitive info such as user names and private key. - [ ] Ensure (1) hyperlinks and markdown anchors are working (2) use relative paths for tutorial repo files (3) put figure and graphs in the `./figure` folder - [ ] Notebook runs automatically `./runner.sh -t <path to .ipynb file>` Signed-off-by: KumoLiu <[email protected]>
1 parent df45e9d commit 422a494

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

performance_profiling/pathology/train_evaluate_nvtx.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -477,6 +477,9 @@ def main(cfg):
477477
# Save final metrics
478478
metric_summary["train_time_per_epoch"] = total_train_time / cfg["n_epochs"]
479479
metric_summary["total_time"] = t_end - t_start
480+
481+
# The type of the value in hparam_dict can only be one of bool, string, float, int, or None.
482+
cfg["grid_shape"] = str(cfg["grid_shape"])
480483
writer.add_hparams(hparam_dict=cfg, metric_dict=metric_summary, run_name=log_dir)
481484
writer.close()
482485
logging.info(f"Metric Summary: {metric_summary}")

0 commit comments

Comments
 (0)