X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=test.py;h=de408aa994dc3224e1949900e89567bf9af48c66;hp=d6d2df1e6584cbdb3b0f3f20833d0fb9e8d8d96f;hb=2cd32038873961c8ff3861efb218fad75fbcbf69;hpb=546ec5453243e2efbf8c3caefdc04394ca87676d

diff --git a/test.py b/test.py
index d6d2df1..de408aa 100755
--- a/test.py
+++ b/test.py
@@ -10,10 +10,11 @@ x = torch.ByteTensor(4, 5).fill_(0)
 
 print(x.size())
 
-mylib.generate_sequence(x)
+mylib.generate_sequence(8, x)
 
 print(x.size())
 
 x = x.float().sub_(128).div_(128)
 
-torchvision.utils.save_image(x[0], 'example.png')
+for s in range(0, x.size(0)):
+    torchvision.utils.save_image(x[s], 'example_' + str(s) + '.png')