mirror of
https://github.com/Shawn-Shan/fawkes.git
synced 2024-11-09 13:41:31 +05:30
app
Former-commit-id: 0282ba3b18eab09351540df5aa9858d466aa3454 [formerly 88ce02e785b35491c0fe9ffec50d9504c6069780] Former-commit-id: d1d8b474c1383b05dca7e3e6084847b6e852c3ad
This commit is contained in:
parent
38f89999fa
commit
448c58ab16
@ -70,7 +70,7 @@ def main(*argv):
|
|||||||
help='GPU id', default='0')
|
help='GPU id', default='0')
|
||||||
|
|
||||||
parser.add_argument('--mode', '-m', type=str,
|
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,
|
parser.add_argument('--feature-extractor', type=str,
|
||||||
help="name of the feature extractor used for optimization",
|
help="name of the feature extractor used for optimization",
|
||||||
default="high_extract")
|
default="high_extract")
|
||||||
@ -95,21 +95,21 @@ def main(*argv):
|
|||||||
args.lr = 20
|
args.lr = 20
|
||||||
elif args.mode == 'mid':
|
elif args.mode == 'mid':
|
||||||
args.feature_extractor = "high_extract"
|
args.feature_extractor = "high_extract"
|
||||||
args.th = 0.004
|
args.th = 0.005
|
||||||
args.max_step = 50
|
args.max_step = 50
|
||||||
args.lr = 15
|
args.lr = 15
|
||||||
elif args.mode == 'high':
|
elif args.mode == 'high':
|
||||||
args.feature_extractor = "high_extract"
|
args.feature_extractor = "high_extract"
|
||||||
args.th = 0.007
|
args.th = 0.008
|
||||||
args.max_step = 100
|
args.max_step = 500
|
||||||
args.lr = 10
|
args.lr = 15
|
||||||
elif args.mode == 'ultra':
|
elif args.mode == 'ultra':
|
||||||
if not tf.test.is_gpu_available():
|
if not tf.test.is_gpu_available():
|
||||||
print("Please enable GPU for ultra setting...")
|
print("Please enable GPU for ultra setting...")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
# args.feature_extractor = ["high_extract", 'high2_extract']
|
# args.feature_extractor = ["high_extract", 'high2_extract']
|
||||||
args.feature_extractor = "high_extract"
|
args.feature_extractor = "high_extract"
|
||||||
args.th = 0.015
|
args.th = 0.01
|
||||||
args.max_step = 2000
|
args.max_step = 2000
|
||||||
args.lr = 8
|
args.lr = 8
|
||||||
elif args.mode == 'custom':
|
elif args.mode == 'custom':
|
||||||
|
Loading…
Reference in New Issue
Block a user