in tensorflow_managed_spot_training_checkpointing/source_dir/cifar10_keras_main.py [0:0]
def num_examples_per_epoch(subset='train'):
if subset == 'train':
return 40000
elif subset == 'validation':
return 10000
elif subset == 'eval':
return 10000
else:
raise ValueError('Invalid data subset "%s"' % subset)