Function random_input_generator

tests/performance_test/vgg/exp_config.py:4–10

(num, batchsize=32, format='NHWC')

def random_input_generator(num, batchsize=32, format='NHWC'):
    input_shape = (batchsize, 224, 224, 3) if format == 'NHWC' else (batchsize, 3, 224, 224)
    rng = np.random.RandomState(1234)
    for i in range(num):
        x = rng.uniform(0.0, 1.0, size=input_shape).astype(np.float32)
        y = rng.randint(0, 1000, size=(batchsize, ))
        yield (x, y)


MONITOR_INTERVAL = 50