diff --git a/training/plot-loss.py b/training/plot-loss.py index eb7d7b3..b201dc0 100755 --- a/training/plot-loss.py +++ b/training/plot-loss.py @@ -59,7 +59,8 @@ def plot(workDirs): plt.ylim(ymin=0) # plt.xlim(xmin=1) plt.grid(b=True, which='major', color='k', linestyle='-') - plt.grid(b=True, which='minor', color='k', linestyle='--', alpha=0.2) + plt.grid(b=True, which='minor', color='k', linestyle='-', alpha=0.2) + plt.minorticks_on() # ax.set_yscale('log') d = os.path.join(plotDir, "train-loss.pdf") fig.savefig(d) @@ -71,6 +72,9 @@ def plot(workDirs): plt.xlabel("Epoch") plt.ylabel("LFW Accuracy") plt.ylim(ymin=0, ymax=1) + plt.grid(b=True, which='major', color='k', linestyle='-') + plt.grid(b=True, which='minor', color='k', linestyle='-', alpha=0.2) + plt.minorticks_on() # plt.xlim(xmin=1) # ax.set_yscale('log') d = os.path.join(plotDir, "lfw-accuracy.pdf")