Seems to work quite well.

[flatland.git] / test.py

diff --git a/test.py b/test.py
index d6d2df1..de408aa 100755 (executable)
--- 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')