diff --git a/trainers/train_pynet.py b/trainers/train_pynet.py index bb167b9..6c99f4b 100644 --- a/trainers/train_pynet.py +++ b/trainers/train_pynet.py @@ -29,7 +29,7 @@ def lr_schedule(epoch, optimizer): elif 10 <= epoch <= 20: lr = 5e-2 elif 25 < epoch <= 45: - lr = 1e-3 + lr = 5e-3 else: lr = 1e-4 diff --git a/util.py b/util.py index 10bf08d..8f517fa 100644 --- a/util.py +++ b/util.py @@ -109,3 +109,5 @@ def save_plot(self): plt.plot(np.arange(len(f2_scores)), f2_scores) plt.savefig('../log/%s_fcscore.jpg' % self.name) + plt.close('all') +