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
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
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
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
def cifar_model_runner(request):
yield base_runner.generate_tnt_model_runner(request.param())