Ejemplos de generate_tnt_model_runner en Python

Ejemplo n.º 1

def gen_model_runners(model_config: base_runner.ModelConfig):
    tnt_model_runner = base_runner.generate_tnt_model_runner(model_config)
    reference_model_runner = base_runner.TrainingRunner(
        model_config.model_generator())
    return tnt_model_runner, reference_model_runner

Ejemplo n.º 2

def model_runners(request):
    tf.random.set_seed(42)
    tnt_model_runner = base_runner.generate_tnt_model_runner(request.param())
    tf.random.set_seed(42)
    reference_model_runner = base_runner.TrainingRunner(request.param())
    yield tnt_model_runner, reference_model_runner

Ejemplo n.º 3

def model_runners(request):
    tnt_model_runner = training_runner.generate_tnt_model_runner(request.param)
    reference_model_runner = training_runner.TrainingRunner(
        request.param.model_generator())
    yield tnt_model_runner, reference_model_runner

Ejemplo n.º 4

Archivo: accuracy_test_any_batch_size.py Proyecto: cc-hpc-itwm/tarantella

def model_runners(request):
    tnt_model_runner = base_runner.generate_tnt_model_runner(
        base_runner.ModelConfig(request.param, tnt.ParallelStrategy.DATA))
    reference_model_runner = base_runner.TrainingRunner(request.param())
    yield tnt_model_runner, reference_model_runner

Ejemplo n.º 5

def cifar_model_runner(request):
    yield base_runner.generate_tnt_model_runner(request.param())