Example #1
0
def test_pytorch():
    yield from utils.check_spawn_with_callback(ExtCallbackTorch)
Example #2
0
def test_mxnet():
    yield from check_spawn_with_callback(ExtCallbackMX)
Example #3
0
def test_tensor_cpu():
    yield from utils.check_spawn_with_callback(utils.ExtCallbackTensorCPU)
Example #4
0
def test_num_outputs():
    yield from utils.check_spawn_with_callback(
        utils.ExtCallbackMultipleOutputs,
        utils.ExtCallbackMultipleOutputs,
        num_outputs=2,
        dtypes=[np.uint8, np.float])
Example #5
0
def test_randomly_shaped_data():
    yield from utils.check_spawn_with_callback(utils.ExtCallback,
                                               shapes=[(100, 40, 3),
                                                       (8, 64, 64, 3)],
                                               random_data=True,
                                               random_shape=True)
Example #6
0
def test_dtypes():
    yield from utils.check_spawn_with_callback(utils.ExtCallback)