From 10df6ded0fa4a6c0c64f0b0afda1d9e2a5a25dec Mon Sep 17 00:00:00 2001 From: leewlving Date: Sat, 15 Jun 2024 20:05:06 +0800 Subject: [PATCH] new update --- main.py | 2 +- utils/get_args.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index e3c978e..36da785 100644 --- a/main.py +++ b/main.py @@ -4,7 +4,7 @@ from train.hash_train import Trainer if __name__ == "__main__": engine=Trainer() - engine.test() + # engine.test() engine.train_epoch() diff --git a/utils/get_args.py b/utils/get_args.py index 42b6a22..99e24ac 100644 --- a/utils/get_args.py +++ b/utils/get_args.py @@ -26,7 +26,7 @@ def get_args(): parser.add_argument("--batch-size", type=int, default=8) parser.add_argument("--num-workers", type=int, default=4) parser.add_argument("--query-num", type=int, default=5120) - parser.add_argument("--train-num", type=int, default=1024) + parser.add_argument("--train-num", type=int, default=128) parser.add_argument("--lr-decay-freq", type=int, default=5) parser.add_argument("--display-step", type=int, default=50) parser.add_argument("--seed", type=int, default=1814)