- if args.compress_vignettes:
- train_set = CompressedVignetteSet(problem_number,
- args.nb_train_samples, args.batch_size,
- cuda = torch.cuda.is_available())
- else:
- train_set = VignetteSet(problem_number,
- args.nb_train_samples, args.batch_size,
- cuda = torch.cuda.is_available())
+ train_set = VignetteSet(problem_number,
+ args.nb_train_samples, args.batch_size,
+ cuda = torch.cuda.is_available())