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