Skip to content

Commit

Permalink
Set output_class = 10 for cifar example
Browse files Browse the repository at this point in the history
  • Loading branch information
BichengYing committed Jun 26, 2020
1 parent 54d41e6 commit 4e56db4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/pytorch_cifar10_resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@
val_dataset, batch_size=args.val_batch_size, sampler=val_sampler, **kwargs
)

model = getattr(models, args.model)()
model = getattr(models, args.model)(num_classes=10)

if args.cuda:
# Move model to GPU.
Expand Down

0 comments on commit 4e56db4

Please sign in to comment.