a1ce72f5e1a49aea9a7e983cd7373f893e3970a6,examples/experiments/profiling/torch_research.py,,,#,56

Before Change

    .add_function(get_accuracy, returns="accuracy", name="test_accuracy", execute="%100", pipeline="test")
)

research.run(n_reps=1, n_iters=1000, workers=4, name="my_research", gpu=[2, 4, 5, 6])

After Change

    .add_function(get_accuracy, returns="accuracy", name="test_accuracy", execute="%100", pipeline="test")
)

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

research.run(n_reps=1, n_iters=1000, workers=n_workers, name="my_research", gpu=gpu_list[:n_workers])

Instances