diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
index e796a99..f892ef0 100644
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -302,12 +302,12 @@ public class DockerTrainService {
 
     // ===== Augmentation =====
     addArg(c, "--rot-prob", req.getRotProb());
-    // addArg(c, "--rot-degree", req.getRotDegree()); // TODO AI 수정되면 주석 해제
+    addArg(c, "--rot-degree", req.getRotDegree());
     addArg(c, "--flip-prob", req.getFlipProb());
     addArg(c, "--exchange-prob", req.getExchangeProb());
     addArg(c, "--brightness-delta", req.getBrightnessDelta());
-    // addArg(c, "--contrast-range", req.getContrastRange()); // TODO AI 수정되면 주석 해제
-    // addArg(c, "--saturation-range", req.getSaturationRange()); // TODO AI 수정되면 주석 해제
+    addArg(c, "--contrast-range", req.getContrastRange());
+    addArg(c, "--saturation-range", req.getSaturationRange());
     addArg(c, "--hue-delta", req.getHueDelta());
 
     addArg(c, "--resume", req.getResumeFrom());