Group setup statements

Toni-SM · Sep 15, 2024 · c15f3ce · c15f3ce
1 parent 4567e80
commit c15f3ce
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/skrl/agents/torch/ppo/ppo.py b/skrl/agents/torch/ppo/ppo.py
@@ -155,10 +155,8 @@ def __init__(self,
 
         self._mixed_precision = self.cfg["mixed_precision"]
 
-        # cuda or cpu
+        # set up automatic mixed precision
         self._device_type = torch.device(device).type
-
-        # set up scaler
         self._scaler = torch.cuda.amp.GradScaler(enabled=self._mixed_precision)
 
         # set up optimizer and learning rate scheduler