+def generate_one_batch(s):
+ svrt.seed(s)
+ target = torch.LongTensor(self.batch_size).bernoulli_(0.5)
+ input = svrt.generate_vignettes(problem_number, target)
+ input = input.float().view(input.size(0), 1, input.size(1), input.size(2))
+ if self.cuda:
+ input = input.cuda()
+ target = target.cuda()
+ return [ input, target ]
+