Esempi in Python per generate_tnt_model_runner

Esempio 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

Esempio 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

Esempio 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

Esempio n. 4

File: accuracy_test_any_batch_size.py Progetto: 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

Esempio n. 5

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