3 changed files with 5 additions and 5 deletions
@@ -5,7 +5,5 @@ source scripts/env.sh
 $DFL_PYTHON "$DFL_SRC" extract \
    --input-dir "$DFL_WORKSPACE/data_dst" \
    --output-dir "$DFL_WORKSPACE/data_dst/aligned" \
-    --multi-gpu \
-    --detector s3fd \
-    --debug-dir "$DFL_WORKSPACE/data_dst/aligned_debug"
+    --detector s3fd 

@@ -6,5 +6,6 @@ $DFL_PYTHON "$DFL_SRC" train \
    --training-data-src-dir "$DFL_WORKSPACE/data_src" \
    --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \
    --model-dir "$DFL_WORKSPACE/model" \
-    --model AMP
+    --model AMP \
+    --cpu-only

@@ -6,4 +6,5 @@ $DFL_PYTHON "$DFL_SRC" train \
    --training-data-src-dir "$DFL_WORKSPACE/data_src" \
    --training-data-dst-dir "$DFL_WORKSPACE/data_dst" \
    --model-dir "$DFL_WORKSPACE/model" \
-    --model DF 
+    --model DF \
+    --cpu-only