From 656d4c1e35b0ca45117c9365fcdb688d7c9682db Mon Sep 17 00:00:00 2001 From: XXM <740560488@qq.com> Date: Wed, 13 Mar 2024 22:29:14 +0800 Subject: [PATCH] change gpu id --- config/cls.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config/cls.yaml b/config/cls.yaml index 4a2d92f..1f83eb0 100644 --- a/config/cls.yaml +++ b/config/cls.yaml @@ -1,7 +1,7 @@ batch_size: 16 epoch: 200 learning_rate: 1e-3 -gpu: 1 +gpu: 0 num_point: 1024 optimizer: Adam weight_decay: 1e-4