|
a |
|
b/dsb2018_topcoders/selim/params.py |
|
|
1 |
import argparse |
|
|
2 |
|
|
|
3 |
parser = argparse.ArgumentParser() |
|
|
4 |
arg = parser.add_argument |
|
|
5 |
arg('--gpu', default="0") |
|
|
6 |
arg('--epochs', type=int, default=100) |
|
|
7 |
arg('--fold', default='0') |
|
|
8 |
arg('--n_folds', type=int, default=4) |
|
|
9 |
arg('--freeze_till_layer', default='input_1') |
|
|
10 |
arg('--preprocessing_function') |
|
|
11 |
arg('--weights') |
|
|
12 |
arg('--learning_rate', type=float, default=0.001) |
|
|
13 |
arg('--crop_size', type=int, default=192) |
|
|
14 |
arg('--crops_per_image', type=int, default=1) |
|
|
15 |
arg('--batch_size', type=int, default=16) |
|
|
16 |
arg('--num_workers', type=int, default=7) |
|
|
17 |
arg('--loss_function', default='bce_dice') |
|
|
18 |
arg('--optimizer', default="rmsprop") |
|
|
19 |
arg('--clr') |
|
|
20 |
arg('--schedule') |
|
|
21 |
arg('--decay', type=float, default=0.0) |
|
|
22 |
arg('--save_period', type=int, default=1) |
|
|
23 |
arg('--network', default='densenet_unet') |
|
|
24 |
arg('--alias', default='') |
|
|
25 |
arg('--steps_per_epoch', type=int, default=0) |
|
|
26 |
arg('--use_softmax', action="store_true") |
|
|
27 |
arg('--use_full_masks', action="store_true") |
|
|
28 |
arg('--multi_gpu', action="store_true") |
|
|
29 |
arg('--seed', type=int, default=777) |
|
|
30 |
arg('--models_dir', default='nn_models') |
|
|
31 |
arg('--images_dir', default='../data/images_all') |
|
|
32 |
arg('--masks_dir', default='../data/masks_all') |
|
|
33 |
arg('--labels_dir', default='../data/labels_all') |
|
|
34 |
arg('--test_folder', default='../data_test') |
|
|
35 |
arg('--folds_csv', default='../data/folds.csv') |
|
|
36 |
arg('--out_root_dir', default='../predictions') |
|
|
37 |
arg('--out_masks_folder') |
|
|
38 |
arg('--models', nargs='+') |
|
|
39 |
arg('--out_channels', type=int, default=2) |
|
|
40 |
|
|
|
41 |
args = parser.parse_args() |