from torchvision import datasets, transforms, utils
-from _ext import svrt
+import svrt
labels = torch.LongTensor(12).zero_()
labels.narrow(0, 0, labels.size(0)//2).fill_(1)
x.div_(255)
torchvision.utils.save_image(x, 'example.png')
+
+print('Wrote example.png')