This commit is contained in:
Eldar Akchurin 2017-02-01 11:40:38 +01:00
Родитель 41f8b7845d
Коммит 85af41723e
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -25,7 +25,7 @@ def test_cifar_resnet_distributed(device_id):
"-q", "32",
"-es", "512",
"-device", "0" ]
mpiexec_test(device_id, script_under_test, params, 0.86, False, 2)
mpiexec_test(device_id, script_under_test, params, 0.86, False, 3)
def test_cifar_resnet_distributed_1bitsgd(device_id):
params = [ "-e", "2",
@ -33,7 +33,7 @@ def test_cifar_resnet_distributed_1bitsgd(device_id):
"-q", "1",
"-es", "512",
"-device", "0" ]
mpiexec_test(device_id, script_under_test, params, 0.86, False, 2)
mpiexec_test(device_id, script_under_test, params, 0.86, False, 3)
def test_cifar_resnet_distributed_block_momentum(device_id):
@ -42,4 +42,4 @@ def test_cifar_resnet_distributed_block_momentum(device_id):
"-b", "3200",
"-es", "512",
"-device", "0" ]
mpiexec_test(device_id, script_under_test, params, 0.89, False, 2)
mpiexec_test(device_id, script_under_test, params, 0.89, False, 5)