def test_cifar_resnet_distributed_block_momentum(device_id):
params = [ "-e", "2",
"-d", data_set_directory(),
"-b", "3200",
"-es", "512",
"-device", "0" ]
mpiexec_test(device_id, script_under_test, params, 0.89, False, 2)
def test_cifar_resnet_distributed_1bitsgd(device_id):
params = [ "-e", "2",
"-d", data_set_directory(),
"-q", "1",
"-es", "512",
"-device", "0" ]
mpiexec_test(device_id, script_under_test, params, 0.86, False, 2)