From 585d00a8fcbe31431f8979259bf851564b24a912 Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 25 Dec 2022 17:59:42 +0100 Subject: [PATCH] development reset --- scripts/5_data_dst_extract_faces_S3FD_all_GPU.sh | 4 +--- scripts/6_train_AMP.sh | 3 ++- scripts/6_train_DF.sh | 3 ++- 3 files changed, 5 insertions(+), 5 deletions(-) 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