development reset

scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh

@@ -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 
 

scripts/6_train_AMP.sh

@@ -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
 

scripts/6_train_DF.sh

@@ -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