예제 #1
0
def test_pytorch():
    yield from utils.check_spawn_with_callback(ExtCallbackTorch)
예제 #2
0
def test_mxnet():
    yield from check_spawn_with_callback(ExtCallbackMX)
예제 #3
0
def test_tensor_cpu():
    yield from utils.check_spawn_with_callback(utils.ExtCallbackTensorCPU)
예제 #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])
예제 #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)
예제 #6
0
def test_dtypes():
    yield from utils.check_spawn_with_callback(utils.ExtCallback)