Commit 58020503 authored by Shanmugam Ramasamy's avatar Shanmugam Ramasamy
Browse files

Merge branch 'better-Output' into 'main'

Better output

See merge request ADLR/megatron-lm!559
parents fdd34a82 2221d882
......@@ -25,6 +25,7 @@ def read_tb_logs_as_list(path, summary_name):
ea.Reload()
summary = ea.Scalars(summary_name)
summary_list = [round(x.value, 5) for x in summary]
print(f'\nObtained the following list for {summary_name} ------------------')
print(summary_list)
return summary_list
raise FileNotFoundError(f"File not found matching: {path}/events*")
......
......@@ -34,6 +34,7 @@ def read_tb_logs_as_list(path, summary_name):
ea.Reload()
summary = ea.Scalars(summary_name)
summary_list = [round(x.value, 5) for x in summary]
print(f'\nObtained the following list for {summary_name} ------------------')
print(summary_list)
return summary_list
raise FileNotFoundError(f"File not found matching: {path}/events*")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment