diff --git a/scripts/6_train_AMP.sh b/scripts/6_train_AMP.sh
index ed372c2..a3400d1 100755
--- a/scripts/6_train_AMP.sh
+++ b/scripts/6_train_AMP.sh
@@ -7,5 +7,5 @@ $DFL_PYTHON "$DFL_SRC" train \
     --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \
     --model-dir "$DFL_WORKSPACE/model" \
     --model AMP \
-    --cpu-only
+    --cpu-only