diff --git a/skrl/agents/torch/ppo/ppo.py b/skrl/agents/torch/ppo/ppo.py
index 4b9f6be2..274d384b 100644
--- 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