コード例 #1
0
def skeleton_autoencoder_imagenet(cluster, dir_name, executables,
                                  compiler_name, weekly):
    if cluster in ['surface', 'pascal']:
        e = 'skeleton_autoencoder_imagenet: does not run on GPU'
        print('Skip - ' + e)
        pytest.skip(e)
    if compiler_name not in executables:
        e = 'skeleton_autoencoder_imagenet: default_exes[%s] does not exist' % compiler_name
        print('Skip - ' + e)
        pytest.skip(e)
    model_folder = 'models/autoencoder_imagenet'
    model_name = 'conv_autoencoder_imagenet'
    should_log = False
    actual_objective_functions = common_code.skeleton(
        cluster,
        dir_name,
        executables[compiler_name],
        model_folder,
        model_name,
        DATA_FIELDS,
        should_log,
        compiler_name=compiler_name,
        weekly=weekly)
    frequency_str = '_nightly'
    if weekly:
        frequency_str = '_weekly'
    run_tests(actual_objective_functions, model_name, dir_name, cluster,
              should_log, compiler_name, frequency_str)
コード例 #2
0
def skeleton_autoencoder_mnist(cluster, dir_name, executables, compiler_name):
  if compiler_name not in executables:
    pytest.skip('default_exes[%s] does not exist' % compiler_name)    
  model_folder = 'models/autoencoder_mnist'
  model_name = 'conv_autoencoder_mnist'
  should_log=False
  actual_objective_functions = common_code.skeleton(cluster, dir_name, executables[compiler_name], model_folder, model_name, DATA_FIELDS, should_log, compiler_name=compiler_name)
  run_tests(actual_objective_functions, model_name, dir_name, cluster, should_log, compiler_name)
コード例 #3
0
def skeleton_performance_lenet_mnist(cluster, dir_name, executables, compiler_name):
  if compiler_name not in executables:
    pytest.skip('default_exes[%s] does not exist' % compiler_name)
  executable = executables[compiler_name]
  model_name = 'lenet_mnist'
  model_folder = 'models/' + model_name
  should_log = False
  actual_performance = common_code.skeleton(cluster, dir_name, executable, model_folder, model_name, DATA_FIELDS, should_log, compiler_name=compiler_name)
  run_tests(actual_performance, model_name, dir_name, should_log, compiler_name, cluster)
コード例 #4
0
def skeleton_performance_alexnet(cluster, dir_name, executables, compiler_name, weekly):
  if compiler_name not in executables:
    pytest.skip('default_exes[%s] does not exist' % compiler_name)
  executable = executables[compiler_name]
  model_name = 'alexnet'
  model_folder = 'models/' + model_name
  should_log = False
  actual_performance = common_code.skeleton(cluster, dir_name, executable, model_folder, model_name, DATA_FIELDS, should_log, compiler_name=compiler_name, weekly=weekly)
  frequency_str = '_nightly'
  if weekly:
    frequency_str = '_weekly'
  run_tests(actual_performance, model_name, dir_name, should_log, compiler_name, cluster, frequency_str)
コード例 #5
0
def skeleton_autoencoder_imagenet(cluster, dir_name, executables, compiler_name, weekly):
  if cluster == 'surface':
    pytest.skip('skeleton_autoencoder_imagenet does not run on surface')
  if compiler_name not in executables:
    pytest.skip('default_exes[%s] does not exist' % compiler_name)
  model_folder = 'models/autoencoder_imagenet'
  model_name = 'conv_autoencoder_imagenet'
  should_log = False
  actual_objective_functions = common_code.skeleton(cluster, dir_name, executables[compiler_name], model_folder, model_name, DATA_FIELDS, should_log, compiler_name=compiler_name, weekly=weekly)
  frequency_str = '_nightly'
  if weekly:
    frequency_str = '_weekly'
  run_tests(actual_objective_functions, model_name, dir_name, cluster, should_log, compiler_name, frequency_str)
コード例 #6
0
def skeleton_autoencoder_mnist(cluster, dir_name, executables, compiler_name):
    if compiler_name not in executables:
        pytest.skip('default_exes[%s] does not exist' % compiler_name)
    model_folder = 'models/autoencoder_mnist'
    model_name = 'conv_autoencoder_mnist'
    should_log = False
    actual_objective_functions = common_code.skeleton(
        cluster,
        dir_name,
        executables[compiler_name],
        model_folder,
        model_name,
        DATA_FIELDS,
        should_log,
        compiler_name=compiler_name)
    run_tests(actual_objective_functions, model_name, dir_name, cluster,
              should_log, compiler_name)
コード例 #7
0
def skeleton_performance_lenet_mnist(cluster, dir_name, executables,
                                     compiler_name):
    if compiler_name not in executables:
        pytest.skip('default_exes[%s] does not exist' % compiler_name)
    executable = executables[compiler_name]
    model_name = 'lenet_mnist'
    model_folder = 'models/' + model_name
    should_log = False
    actual_performance = common_code.skeleton(cluster,
                                              dir_name,
                                              executable,
                                              model_folder,
                                              model_name,
                                              DATA_FIELDS,
                                              should_log,
                                              compiler_name=compiler_name)
    run_tests(actual_performance, model_name, dir_name, should_log,
              compiler_name, cluster)
コード例 #8
0
def skeleton_performance_alexnet(cluster, dir_name, executables, compiler_name,
                                 weekly):
    if compiler_name not in executables:
        pytest.skip('default_exes[%s] does not exist' % compiler_name)
    executable = executables[compiler_name]
    model_name = 'alexnet'
    model_folder = 'models/' + model_name
    should_log = False
    actual_performance = common_code.skeleton(cluster,
                                              dir_name,
                                              executable,
                                              model_folder,
                                              model_name,
                                              DATA_FIELDS,
                                              should_log,
                                              compiler_name=compiler_name,
                                              weekly=weekly)
    frequency_str = '_nightly'
    if weekly:
        frequency_str = '_weekly'
    run_tests(actual_performance, model_name, dir_name, should_log,
              compiler_name, cluster, frequency_str)