params.py

import argparse
import distutils.util

parser = argparse.ArgumentParser()
arg = parser.add_argument
#arg('--gpu', default="0")
arg('--log-dir', default='./logs', help='tensorboard log directory')
arg('--fp16-allreduce', action='store_true', default=False,
                    help='use fp16 compression during allreduce')
arg('--fold', type=int, default=None)
arg('--n_folds', type=int, default=5)
arg('--folds_source')
arg('--clr')
arg('--checkpoint-format', default='./checkpoint-{epoch}.h5',
                    help='checkpoint file format')
arg('--seed', type=int, default=80)
arg('--test_size_float', type=float, default=0.1)
arg('--epochs', type=int, default=30)
arg('--img_height', type=int, default=256)
arg('--img_width', type=int, default=256)
arg('--out_height', type=int, default=256)
arg('--out_width', type=int, default=256)
arg('--input_width', type=int, default=256)
arg('--input_height', type=int, default=256)
arg('--use_crop', type=distutils.util.strtobool, default='true')
arg('--learning_rate', type=float, default=0.0125)
arg('--batch_size', type=int, default=1)
arg('--val-batch-size', type=int, default=4,
                    help='input batch size for validation')
arg('--dataset_dir', default='/data/user/wsmonroe/MetalsGroup/pennyProject/pennyData/TrainingSet')
arg('--models_dir', default='models')
arg('--weights')
arg('--loss_function', default='bce_dice')
arg('--freeze_till_layer', default='input_1')
arg('--show_summary', type=bool)
arg('--network', default='resnet50')
arg('--net_alias', default='')
arg('--preprocessing_function', default='caffe')
arg('--base-lr', type=float, default=0.0125,
                    help='learning rate for a single GPU')
arg('--warmup-epochs', type=float, default=5,
                    help='number of warmup epochs')
arg('--momentum', type=float, default=0.9,
                    help='SGD momentum')
arg('--wd', type=float, default=0.00005,
                    help='weight decay')

arg('--pred_mask_dir')
arg('--pred_tta')
arg('--pred_batch_size', default=1)
arg('--test_data_dir', default='/data/user/wsmonroe/MetalsGroup/pennyProject/data/images/chopped')
arg('--pred_threads', type=int, default=1)
arg('--submissions_dir', default='submissions')
arg('--pred_sample_csv', default='input/sample_submission.csv')
arg('--predict_on_val', type=bool, default=False)
arg('--stacked_channels', type=int, default=0)
arg('--stacked_channels_dir', default=None)

# Dir names
arg('--train_data_dir_name', default='train')
arg('--val_data_dir_name', default='train')
arg('--train_mask_dir_name', default='train_masks')
arg('--val_mask_dir_name', default='train_masks')

arg('--dirs_to_ensemble', nargs='+')
arg('--ensembling_strategy', default='average')
arg('--folds_dir')
arg('--ensembling_dir')
arg('--ensembling_cpu_threads', type=int, default=6)

args = parser.parse_args()