Commit 2221d882 authored by Shanmugam Ramasamy's avatar Shanmugam Ramasamy
Browse files

Better output

parent fdd34a82
...@@ -25,6 +25,7 @@ def read_tb_logs_as_list(path, summary_name): ...@@ -25,6 +25,7 @@ def read_tb_logs_as_list(path, summary_name):
ea.Reload() ea.Reload()
summary = ea.Scalars(summary_name) summary = ea.Scalars(summary_name)
summary_list = [round(x.value, 5) for x in summary] summary_list = [round(x.value, 5) for x in summary]
print(f'\nObtained the following list for {summary_name} ------------------')
print(summary_list) print(summary_list)
return summary_list return summary_list
raise FileNotFoundError(f"File not found matching: {path}/events*") raise FileNotFoundError(f"File not found matching: {path}/events*")
......
...@@ -34,6 +34,7 @@ def read_tb_logs_as_list(path, summary_name): ...@@ -34,6 +34,7 @@ def read_tb_logs_as_list(path, summary_name):
ea.Reload() ea.Reload()
summary = ea.Scalars(summary_name) summary = ea.Scalars(summary_name)
summary_list = [round(x.value, 5) for x in summary] summary_list = [round(x.value, 5) for x in summary]
print(f'\nObtained the following list for {summary_name} ------------------')
print(summary_list) print(summary_list)
return summary_list return summary_list
raise FileNotFoundError(f"File not found matching: {path}/events*") 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