diff --git a/config/config.py b/config/config.py index bd022bed2184fc2a6f5d14db3000c9a6e7a7c58a..068b2c5ed0f51cbc09e3cd646c02ba4d6c9cd2fe 100644 --- a/config/config.py +++ b/config/config.py @@ -3,7 +3,7 @@ import argparse def load_args(): parser = argparse.ArgumentParser() - parser.add_argument('--epoches', type=int, default=50) + parser.add_argument('--epoches', type=int, default=20) parser.add_argument('--eval_inter', type=int, default=1) parser.add_argument('--augment_args', nargs = '+', type = float, default = [0,0,0,0.1,0.1,0.1,0.1]) parser.add_argument('--noise_threshold', type=int, default=0)