From f52d11c7dff4721cc1f2d3b9f2bc3b39cc006d89 Mon Sep 17 00:00:00 2001 From: Charles Gaydon <11660435+CharlesGaydon@users.noreply.github.com> Date: Tue, 2 Jan 2024 17:42:29 +0100 Subject: [PATCH] Update RandLaNet_base_run_FR-2x3GPUs.yaml Prefer using trainer.devices to trainer.gpus --- configs/experiment/RandLaNet_base_run_FR-2x3GPUs.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/experiment/RandLaNet_base_run_FR-2x3GPUs.yaml b/configs/experiment/RandLaNet_base_run_FR-2x3GPUs.yaml index 888ec903..03c8511e 100755 --- a/configs/experiment/RandLaNet_base_run_FR-2x3GPUs.yaml +++ b/configs/experiment/RandLaNet_base_run_FR-2x3GPUs.yaml @@ -14,5 +14,5 @@ trainer: strategy: ddp_find_unused_parameters_false accelerator: gpu num_nodes: 2 - gpus: 3 + devices: 3 accumulate_grad_batches: 1