update

2017-05-23 10:37:51 +10:00
parent 30c0070497
commit d87529ed89
10 changed files with 17 additions and 10 deletions
--- a/tutorial-contents/404_autoencoder.py
+++ b/tutorial-contents/404_autoencoder.py
@ -85,7 +85,6 @@ loss_func = nn.MSELoss()
 # initialize figure
 f, a = plt.subplots(2, N_TEST_IMG, figsize=(5, 2))
 plt.ion()   # continuously plot
-plt.show()

 # original data (first row) for viewing
 view_data = Variable(train_data.train_data[:N_TEST_IMG].view(-1, 28*28).type(torch.FloatTensor)/255.)