fix dupliate gpu check (#9578)

dmlc · Sep 13, 2023 · a343ae3 · a343ae3
1 parent 300f9ac
commit a343ae3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/demo/nvflare/vertical/custom/trainer.py b/demo/nvflare/vertical/custom/trainer.py
@@ -83,9 +83,8 @@ def _do_training(self, fl_ctx: FLContext):
                 'eval_metric': 'auc',
             }
             if self._use_gpus:
-                if self._use_gpus:
-                    self.log_info(fl_ctx, f'Training with GPU {rank}')
-                    param['device'] = f"cuda:{rank}"
+                self.log_info(fl_ctx, f'Training with GPU {rank}')
+                param['device'] = f"cuda:{rank}"
 
             # specify validations set to watch performance
             watchlist = [(dtest, "eval"), (dtrain, "train")]