-
Notifications
You must be signed in to change notification settings - Fork 101
/
Copy pathargs.py
69 lines (55 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
66
67
68
69
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=30, type=int)
# model
parser.add_argument('--batch_size', default=160, 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=1048, 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=720, type=int)
parser.add_argument('--start_epoch', default=0, type=int)
parser.add_argument('--ingrW2V', default='data/text/vocab.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('--resume', default='', type=str)
# test
parser.add_argument('--path_results', default='results/', type=str)
parser.add_argument('--model_path', default='snapshots/model_e220_v-4.700.pth.tar', type=str)
parser.add_argument('--test_image_path', default='chicken.jpg', 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 = 'vocab.txt', type=str)
parser.add_argument('--dataset', default = '../data/recipe1M/', type=str)
parser.add_argument('--sthdir', default = '../data/', type=str)
return parser