diff --git a/utils/utils.py b/utils/utils.py index 17a949e8dc..2b5f34cb98 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -1041,7 +1041,7 @@ def plot_results(start=0, stop=0, bucket='', id=()): # from utils.utils import 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].plot(x, y, marker='.', label=Path(f).stem, linewidth=2, markersize=8) ax[i].set_title(s[i]) 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])