DECAPS/config.py at master · hula-ai/DECAPS · GitHub

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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from optparse import OptionParser


parser = OptionParser()

parser.add_option('-e', '--epochs', dest='epochs', default=500, type='int',
                  help='number of epochs (default: 80)')
parser.add_option('-b', '--batch-size', dest='batch_size', default=1, type='int',
                  help='batch size (default: 16)')
parser.add_option('--df', '--disp_freq', dest='disp_freq', default=2, type='int',
                  help='frequency of displaying the training results (default: 100)')
parser.add_option('--vf', '--val_freq', dest='val_freq', default=5, type='int',
                  help='run validation for each <val_freq> iterations (default: 2000)')
parser.add_option('-j', '--workers', dest='workers', default=0, type='int',
                  help='number of data loading workers (default: 16)')
# For data
parser.add_option('--dn', '--data_name', dest='data_name', default='chexpert',
                  help='mnist, fashion_mnist, t_mnist, c_mnist, chexpert, rsna, mias, deeplesion (default: mnist)')

parser.add_option('--ih', '--img_h', dest='img_h', default=512, type='int',
                  help='input image height (default: 28)')
parser.add_option('--iw', '--img_w', dest='img_w', default=512, type='int',
                  help='input image width (default: 28)')
parser.add_option('--ic', '--img_c', dest='img_c', default=3, type='int',
                  help='number of input channels (default: 1)')
parser.add_option('--nc', '--num_classes', dest='num_classes', default=2, type='int',
                  help='number of classes (default: 10)')

parser.add_option('--ni', '--num_iterations', dest='num_iterations', default=3, type='int',
                  help='number of routing iterations (default: 3)')

# Pick the loss type
parser.add_option('--lt', '--loss_type', dest='loss_type', default='margin',
                  help='margin, spread, cross-entropy (default: margin)')

# For Margin loss
parser.add_option('--mp', '--m_plus', dest='m_plus', default=0.9, type='float',
                  help='m+ parameter (default: 0.9)')
parser.add_option('--mm', '--m_minus', dest='m_minus', default=0.1, type='float',
                  help='m- parameter (default: 0.1)')
parser.add_option('--la', '--lambda_val', dest='lambda_val', default=0.5, type='float',
                  help='Down-weighting parameter for the absent class (default: 0.5)')

# For Spread loss
parser.add_option('--mmin', '--m_min', dest='m_min', default=0.2, type='float',
                  help='m_min parameter (default: 0.2)')
parser.add_option('--mmax', '--m_max', dest='m_max', default=0.9, type='float',
                  help='m_min parameter (default: 0.9)')
parser.add_option('--nefm', '--n_eps_for_m', dest='n_eps_for_m', default=1, type='int',
                  help='number of epochs to increment the margin (default: 5)')
parser.add_option('--md', '--m_delta', dest='m_delta', default=0.1, type='float',
                  help='margin increment (default: 0.1)')

# For Reconstructuion loss
parser.add_option('--al', '--alpha', dest='alpha', default=0.0005, type='float',
                  help='Regularization coefficient to scale down the reconstruction loss (default: 0.0005)')

# For Attention Regularization Loss
parser.add_option('--ar', '--attention_reg', dest='attention_reg', default=True,
                  help='whether to use attention regularization loss or not (default: True)')
parser.add_option('--laone', '--lambda_one', dest='lambda_one', default=1, type='float',
                  help='Attention regularization coefficient (default: 10e7)')

# For optimizer
parser.add_option('--lr', '--lr', dest='lr', default=0.00001, type='float',
                  help='learning rate (default: 0.001)')
parser.add_option('--beta1', '--beta1', dest='beta1', default=0.5, type='float',
                  help='beta 1 for Adam optimizer (default: 0.9)')


# For CapsNet
parser.add_option('--fe', '--feature_extractor', dest='feature_extractor', default='inception',
                  help='densenet, resnet, inception, custom (default: resnet)')

parser.add_option('--f1', '--f1', dest='f1', default=512, type='int',
                  help='number of filters for the conv1 layer (default: 256)')
parser.add_option('--k1', '--k1', dest='k1', default=9, type='int',
                  help='filter size of the conv1 layer (default: 9)')
parser.add_option('--s1', '--s1', dest='s1', default=1, type='int',
                  help='stride of the conv1 layer (default: 1)')

parser.add_option('--f2', '--f2', dest='f2', default=256, type='int',
                  help='number of filters for the primary capsule layer (default: 256)')
parser.add_option('--k2', '--k2', dest='k2', default=9, type='int',
                  help='filter size of the primary capsule layer (default: 9)')

parser.add_option('--pcd', '--primary_cap_dim', dest='primary_cap_dim', default=64, type='int',
                  help='dimension of each primary capsule (default: 8)')
parser.add_option('--dcd', '--digit_cap_dim', dest='digit_cap_dim', default=16, type='int',
                  help='dimension of each digit capsule (default: 16)')

# For decoder
parser.add_option('--ad', '--add_decoder', dest='add_decoder', default=False,
                  help='whether to use decoder or not')
parser.add_option('--h1', '--h1', dest='h1', default=512, type='int',
                  help='number of hidden units of the first hidden layer (default: 512)')
parser.add_option('--h2', '--h2', dest='h2', default=1024, type='int',
                  help='number of hidden units of the first hidden layer (default: 1024)')

# For Options
parser.add_option('--ws', '--weight_share', dest='share_weight', default=True,
                  help='whether to share W among child capsules of the same type (default: True)')

parser.add_option('--ca', '--add_coord', dest='add_coord', default=False,
                  help='whether to add coordinates to the primary capsules output or not (default: False)')
parser.add_option('--ncc', '--norm_coord', dest='norm_coord', default=False,
                  help='whether to normalize the coordinates in [0, 1] or not (default: False)')

parser.add_option('--usn', '--use_simnet', dest='use_simnet', default=False,
                  help='whether to use the Similarity Network or not (default: False)')

parser.add_option('--sd', '--save-dir', dest='save_dir', default='./save',
                  help='saving directory of .ckpt models (default: ./save)')

parser.add_option('--lp', '--load_model_path', dest='load_model_path',
                  default='/home/.../models/18122.ckpt',
                  help='path to load a .ckpt model')

options, _ = parser.parse_args()