33885f74bb5ca143c50f6f09c8e16776cf7a5057,examples/experiments/profiling/tf_research.py,,,#,63

Before Change

    metrics = pipeline.get_variable("metrics")
    return metrics.evaluate("accuracy")

research = (Research()
    .add_pipeline(root=train_root, branch=train_template, variables="loss", name="train")
    .add_pipeline(root=test_root, branch=test_template, name="test", run=True, execute="%100", import_from="train")
    .add_grid(grid)
    .add_function(get_accuracy, returns="accuracy", name="test_accuracy", execute="%100", pipeline="test")
)

After Change

n_workers = 1 if len(sys.argv) <= 1 else int(sys.argv[1])
gpu_list = [2, 4, 5, 6]

research.run(n_reps=8, n_iters=1000, workers=n_workers, name="tf_research_"+str(n_workers), gpu=gpu_list[:n_workers])

Instances