From: Francois Fleuret 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)