This commit is contained in:
Glenn Jocher 2020-01-14 22:11:09 -08:00
parent 78ac3bdcfb
commit c6b44befde
1 changed files with 1 additions and 0 deletions

View File

@ -1027,6 +1027,7 @@ def plot_results(start=0, stop=0, bucket='', id=()): # from utils.utils import
y = results[i, x]
if i in [0, 1, 2, 5, 6, 7]:
y[y == 0] = np.nan # dont show zero loss values
# y /= y[0] # normalize
ax[i].plot(x, y, marker='.', label=Path(f).stem)
ax[i].set_title(s[i])
if i in [5, 6, 7]: # share train and val loss y axes