diff --git a/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh b/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh index b407d24..d61c34d 100755 --- a/scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh +++ b/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 diff --git a/scripts/6_train_AMP.sh b/scripts/6_train_AMP.sh index 02f262e..ed372c2 100755 --- a/scripts/6_train_AMP.sh +++ b/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 diff --git a/scripts/6_train_DF.sh b/scripts/6_train_DF.sh index e698178..6e63d3b 100755 --- a/scripts/6_train_DF.sh +++ b/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