def setUp(self):
super(Cifar100Test, self).setUp(
data_wrapper=cifar.CifarData(num_classes=100),
num_classes=100,
expected_num_samples=dict(
train=45000,
val=5000,
trainval=50000,
test=10000,
),
required_tensors_shapes={
"image": (32, 32, 3),
"label": (),
})
def test_incorrect_classes(self):
with self.assertRaisesWithLiteralMatch(
ValueError, "Number of classes must be 10 or 100, got 99"):
cifar.CifarData(num_classes=99)