From 448c58ab163ddd9f9bb8b30c8d087f5a8a2d7cda Mon Sep 17 00:00:00 2001 From: Shawn-Shan Date: Wed, 8 Jul 2020 23:10:13 -0500 Subject: [PATCH] app Former-commit-id: 0282ba3b18eab09351540df5aa9858d466aa3454 [formerly 88ce02e785b35491c0fe9ffec50d9504c6069780] Former-commit-id: d1d8b474c1383b05dca7e3e6084847b6e852c3ad --- fawkes/protection.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/fawkes/protection.py b/fawkes/protection.py index 209985f..9068568 100644 --- a/fawkes/protection.py +++ b/fawkes/protection.py @@ -70,7 +70,7 @@ def main(*argv): help='GPU id', default='0') parser.add_argument('--mode', '-m', type=str, - help='cloak generation mode', default='mid') + help='cloak generation mode', default='low') parser.add_argument('--feature-extractor', type=str, help="name of the feature extractor used for optimization", default="high_extract") @@ -95,21 +95,21 @@ def main(*argv): args.lr = 20 elif args.mode == 'mid': args.feature_extractor = "high_extract" - args.th = 0.004 + args.th = 0.005 args.max_step = 50 args.lr = 15 elif args.mode == 'high': args.feature_extractor = "high_extract" - args.th = 0.007 - args.max_step = 100 - args.lr = 10 + args.th = 0.008 + args.max_step = 500 + args.lr = 15 elif args.mode == 'ultra': if not tf.test.is_gpu_available(): print("Please enable GPU for ultra setting...") sys.exit(1) # args.feature_extractor = ["high_extract", 'high2_extract'] args.feature_extractor = "high_extract" - args.th = 0.015 + args.th = 0.01 args.max_step = 2000 args.lr = 8 elif args.mode == 'custom':