+for arg in vars(args):
+ log_string('argument ' + str(arg) + ' ' + str(getattr(args, arg)))
+
+for problem_number in range(1, 24):
+ if args.compress_vignettes:
+ train_set = CompressedVignetteSet(problem_number, args.nb_train_batches, args.batch_size,
+ cuda=torch.cuda.is_available())
+ test_set = CompressedVignetteSet(problem_number, args.nb_test_batches, args.batch_size,
+ cuda=torch.cuda.is_available())
+ else:
+ train_set = VignetteSet(problem_number, args.nb_train_batches, args.batch_size,
+ cuda=torch.cuda.is_available())
+ test_set = VignetteSet(problem_number, args.nb_test_batches, args.batch_size,
+ cuda=torch.cuda.is_available())