From: Francois Fleuret <francois.fleuret@idiap.ch>
Date: Fri, 4 Jan 2019 14:34:03 +0000 (+0100)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=f07dc15e422fd58a38c5a2ea3b260cd2b44e21af;p=pytorch.git

Update.
---

diff --git a/mine_mnist.py b/mine_mnist.py
index 1d69640..f8b859d 100755
--- a/mine_mnist.py
+++ b/mine_mnist.py
@@ -86,7 +86,6 @@ parser.add_argument('--learning_rate',
 parser.add_argument('--independent', action = 'store_true',
                     help = 'Should the pair components be independent')
 
-
 ######################################################################
 
 args = parser.parse_args()
@@ -359,6 +358,7 @@ elif args.data == 'sequence_pair':
     create_pairs = create_sequences_pairs
     model = NetForSequencePair()
 
+    ######################
     ## Save for figures
     a, b, c = create_pairs()
     for k in range(10):
@@ -366,6 +366,7 @@ elif args.data == 'sequence_pair':
         for i in range(a.size(1)):
             file.write(f'{a[k, i]:f} {b[k,i]:f}\n')
         file.close()
+    ######################
 
 else:
     raise Exception('Unknown data ' + args.data)