def __init__(self): self.submission_data = ChallengeData() self.solution_data = ChallengeData()
help='disables CUDA training') args = parser.parse_args() args.cuda = not args.no_cuda and torch.cuda.is_available() # Load CIFAR10 using torch data paradigm kwargs = {'num_workers': 1, 'pin_memory': True} if args.cuda else {} cifar10_mean_color = [0.49131522, 0.48209435, 0.44646862] # std dev of color across training images cifar10_std_color = [0.01897398, 0.03039277, 0.03872553] transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize(cifar10_mean_color, cifar10_std_color), ]) test_dataset = ChallengeData(args.test_dir, download=True, transform=transform) # Datasets test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=args.test_batch_size, shuffle=False, **kwargs) if os.path.exists(args.model): model = torch.load(args.model) model.drop_path_prob = 0.0 else: print('Model path specified does not exst') sys.exit(1) # cross-entropy loss function criterion = F.cross_entropy if args.cuda: model.cuda()