We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3e41e15 commit b445cefCopy full SHA for b445cef
src/thunder/utils/utils.py
@@ -1,4 +1,5 @@
1
import json
2
+import logging
3
import numpy as np
4
import os
5
import random
@@ -172,8 +173,10 @@ def save_outputs(res_folder: str, outputs: dict) -> None:
172
173
:param res_folder: path where to save outputs.
174
:param outputs: dictionary of outputs.
175
"""
- with open(os.path.join(res_folder, "outputs.json"), "w") as outfile:
176
+ output_filename = os.path.join(res_folder, "outputs.json")
177
+ with open(output_filename, "w") as outfile:
178
json.dump(outputs, outfile)
179
+ logging.info(f"Outputs saved at {output_filename}")
180
181
182
def set_seed(seed: int) -> None:
0 commit comments