"...resnet50_tensorflow.git" did not exist on "b363df84e69269dd9a0b7b439ea32d788e1dd97e"
Unverified Commit 810c44d8 authored by Ziyi Wu's avatar Ziyi Wu Committed by GitHub
Browse files

[Fix] Fix potential bug in using `analyze_logs.py` (#720)

* fix bug when interval == 1

* minor fix
parent 6cd23071
...@@ -53,10 +53,18 @@ def plot_curve(log_dicts, args): ...@@ -53,10 +53,18 @@ def plot_curve(log_dicts, args):
f'{args.json_logs[i]} does not contain metric {metric}') f'{args.json_logs[i]} does not contain metric {metric}')
if args.mode == 'eval': if args.mode == 'eval':
if min(epochs) == args.interval:
x0 = args.interval
else:
# if current training is resumed from previous checkpoint # if current training is resumed from previous checkpoint
# we lost information in early epochs # we lost information in early epochs
# `xs` should start according to `min(epochs)` # `xs` should start according to `min(epochs)`
x0 = min(epochs) + args.interval - min(epochs) % args.interval if min(epochs) % args.interval == 0:
x0 = min(epochs)
else:
# find the first epoch that do eval
x0 = min(epochs) + args.interval - \
min(epochs) % args.interval
xs = np.arange(x0, max(epochs) + 1, args.interval) xs = np.arange(x0, max(epochs) + 1, args.interval)
ys = [] ys = []
for epoch in epochs[args.interval - 1::args.interval]: for epoch in epochs[args.interval - 1::args.interval]:
......
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