2d9caa8a3180a9cbccf9c4bf409e2e7a89bf045f,ml/rl/workflow/dqn_workflow.py,,main,#Any#,145

Before Change

def main(params):
    if params["use_all_avail_gpus"]:
        params["num_gpus"] = torch.cuda.device_count()
        multiprocessing.spawn(
            single_process_main, nprocs=params["num_gpus"], args=[params]
        )
    else:

After Change

        )
        params["use_all_avail_gpus"] = False
    if params["use_all_avail_gpus"]:
        params["num_processes_per_node"] = max(1, torch.cuda.device_count())
        multiprocessing.spawn(
            single_process_main, nprocs=params["num_processes_per_node"], args=[params]
        )
    else:

Instances