Pytorch 如何使用AX运行并行测试？

根据AX文档（），在服务API下，它说“可以并行评估试验”。我如何实现这一点

我有4个GPU，我想一次评估4组不同的超参数（同时进行4次训练）

---

你知道怎么做吗？@nbubis这可能会有帮助：。我跟着导游走了，就在今天，我做了一些事情并开始工作。如果不使用Ray，我无法进行并行工作。你知道怎么做吗？@nbubis这可能会有帮助：。我跟着导游走了，就在今天，我做了一些事情并开始工作。如果不使用Ray，我无法进行并行工作。

from ax.service.ax_client import AxClient
from ax.utils.measurement.synthetic_functions import branin

ax = AxClient()
ax.create_experiment(
    name="branin_test_experiment",
    parameters=[
        {
            "name": "x1",
            "type": "range",
            "bounds": [-5.0, 10.0],
            "value_type": "float",
        },
        {
            "name": "x2",
            "type": "range",
            "bounds": [0.0, 10.0],
        },
    ],
    objective_name="branin",
    minimize=True,
)

for _ in range(15):
    parameters, trial_index = ax.get_next_trial()
    ax.complete_trial(trial_index=trial_index, raw_data=branin(parameters["x1"], parameters["x2"]))

best_parameters, metrics = ax.get_best_parameters()