liwenyun
/
advclip
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new update

This commit is contained in:
leewlving 2024-06-15 20:05:06 +08:00
parent c655bb59e9
commit 10df6ded0f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -4,7 +4,7 @@ from train.hash_train import Trainer
if __name__ == "__main__":
engine=Trainer()
engine.test()
# engine.test()
engine.train_epoch()

2
utils/get_args.py
View File

@ -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)