This commit is contained in:
Glenn Jocher 2019-12-10 20:02:58 -08:00
parent 9f24c12c14
commit a6f87a28e7
1 changed files with 1 additions and 1 deletions

View File

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