fpnet62_wd_1e-4_sgd

ajing · Jun 7, 2017 · 728fb68 · 728fb68
1 parent 5a49c2a
commit 728fb68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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')
+
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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')