Example #1
0
def test_sequence_to_sequence_distributed_block_momentum(device_id):
    params = [ "-e", "4",
               "-datadir", cmudict_dataset_directory(),
               "-ms", "100",
               "-es", "1000",
               "-b", "3200",
               "-device", str(device_id) ]
    mpiexec_test(device_id, script_under_test, mpiexec_params, params, 0.8612, False, 1, 2E-2)
Example #2
0
def test_sequence_to_sequence_distributed_gpu(device_id):
    params = [ "-e", "2",
               "-datadir", cmudict_dataset_directory(),
               "-q", "1",
               "-ms", "100",
               "-es", "500",
               "-device", str(device_id) ]
    mpiexec_test(device_id, script_under_test, mpiexec_params, params, 0.8625, False, 0, 2E-2)