Код: Выделить всё
@ray.remote
def run_experiment(...):
(...)
if __name__ == '__main__':
ray.init()
exp_config = sys.argv[1]
params_tuples, num_cpus, num_gpus = load_exp_config(exp_config)
ray.get([run_experiment.options(num_cpus=num_cpus,
num_gpus=num_gpus).remote(*args) for args in params_tuples])
Код: Выделить всё
@ray.remote(num_cpus=num_cpus, num_gpus=num_gpus)
def run_experiment(...):
(...)
if __name__ == '__main__':
ray.init()
exp_config = sys.argv[1]
params_tuples, _, _ = load_exp_config(exp_config)
ray.get([run_experiment.remote(*args) for args in params_tuples])
Подробнее здесь: https://stackoverflow.com/questions/798 ... nt-to-sett
Мобильная версия