diff --git a/image_ref/config.py b/image_ref/config.py index d0d52db627854b4c2283a1db5f932c7e15314353..32d0d6f10a1d74764204c0f9c1c05e1d6be6e09d 100644 --- a/image_ref/config.py +++ b/image_ref/config.py @@ -5,8 +5,8 @@ def load_args_contrastive(): parser = argparse.ArgumentParser() parser.add_argument('--epoches', type=int, default=100) - parser.add_argument('--save_inter', type=int, default=50) parser.add_argument('--eval_inter', type=int, default=1) + parser.add_argument('--eval_inter', type=int, default=10) parser.add_argument('--noise_threshold', type=int, default=500) parser.add_argument('--lr', type=float, default=0.001) parser.add_argument('--batch_size', type=int, default=64)