-
Notifications
You must be signed in to change notification settings - Fork 1
/
args.py
69 lines (56 loc) · 3.07 KB
/
args.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import argparse
def get_parser():
parser = argparse.ArgumentParser(description='tri-joint parameters')
# general
parser.add_argument('--seed', default=1234, type=int)
parser.add_argument('--no-cuda', action='store_true')
# data
parser.add_argument('--img_path', default='../../data/images/')
parser.add_argument('--data_path', default='../../data/')
parser.add_argument('--workers', default=10, type=int)
# model
parser.add_argument('--batch_size', default=100, type=int)
parser.add_argument('--snapshots', default='../../snapshots/',type=str)
# im2recipe model
parser.add_argument('--embDim', default=1024, type=int)
parser.add_argument('--nRNNs', default=1, type=int)
parser.add_argument('--srnnDim', default=1024, type=int)
parser.add_argument('--irnnDim', default=300, type=int)
parser.add_argument('--imfeatDim', default=2048, type=int)
parser.add_argument('--stDim', default=1024, type=int)
parser.add_argument('--ingrW2VDim', default=300, type=int)
parser.add_argument('--maxSeqlen', default=20, type=int)
parser.add_argument('--maxIngrs', default=20, type=int)
parser.add_argument('--maxImgs', default=5, type=int)
parser.add_argument('--numClasses', default=1005, type=int)
parser.add_argument('--preModel', default='resNet50',type=str)
parser.add_argument('--semantic_reg', default=True,type=bool)
# parser.add_argument('--semantic_reg', default=False,type=bool)
# training
parser.add_argument('--lr', default=0.0001, type=float)
parser.add_argument('--momentum', default=0.9, type=float)
parser.add_argument('--weight_decay', default=0, type=float)
parser.add_argument('--epochs', default=500, type=int)
parser.add_argument('--start_epoch', default=0, type=int)
parser.add_argument('--ingrW2V', default='../../data/text/vocab_v4.bin',type=str)
parser.add_argument('--valfreq', default=10,type=int)
parser.add_argument('--patience', default=1, type=int)
parser.add_argument('--freeVision', default=False, type=bool)
parser.add_argument('--freeRecipe', default=True, type=bool)
parser.add_argument('--cos_weight', default=0.98, type=float)
parser.add_argument('--cls_weight', default=0.01, type=float)
parser.add_argument('--gama', default=16, type=int)
parser.add_argument('--margin', default=0.0, type=float)
parser.add_argument('--resume', default='', type=str)
# test
parser.add_argument('--path_results', default='../../results/', type=str)
parser.add_argument('--result_dir', default='SEJE-resnext101', type=str)
# MedR / Recall@1 / Recall@5 / Recall@10
parser.add_argument('--embtype', default='image', type=str) # [image|recipe] query type
parser.add_argument('--medr', default=1000, type=int)
# dataset
parser.add_argument('--maxlen', default=20, type=int)
parser.add_argument('--vocab', default = '../../data/text/vocab_v4.txt', type=str)
parser.add_argument('--dataset', default = '../../data/recipe1M/', type=str)
parser.add_argument('--sthdir', default = '../../data/', type=str)
return parser