configs.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
@File  : all_configs.py
@Author: Jinkai Zheng
@Date  : 2019/12/30 20:09
@E-mail  : zhengjinkai3@qq.com
'''

import argparse

def parser_argument():
    parser = argparse.ArgumentParser()
    parser.add_argument('--cfgs', type=str, nargs='*',
                            help='config files to load')

    # lr policy to be used
    parser.add_argument('--lr-policy', default='step', type=str,
                     help='lr policy to be used. (default: step)')

    # args for network training
    parser.add_argument('--max-epoch', default=300, type=int,
                     help='max epoch per round. (default: 200)')
    parser.add_argument('--max-round', default=5, type=int,
                     help='max iteration, including initialisation one. '
                          '(default: 5)')
    parser.add_argument('--iter-size', default=1, type=int,
                     help='caffe style iter size. (default: 1)')
    parser.add_argument('--display-freq', default=20, type=int,
                     help='display step')
    parser.add_argument('--gpus', default='', type=str,
                        help='available gpu list. (default: \'\')')
    parser.add_argument('--source', default='casia-b', type=str,
                        help='source dataset to be used. (default: casia-b)')
    parser.add_argument('--target', default='oulp', type=str,
                        help='target dataset to be used. (default: oulp)')
    parser.add_argument('--ANs-select-rate', default=0.25, type=float)
    parser.add_argument('--loss-init-rate', default=0.1, type=float)
    parser.add_argument('--ANs-size', default=1, type=int)
    parser.add_argument('--log-name', default='logloglog', type=str)
    parser.add_argument('--resize', default=32)
    parser.add_argument('--start-round', default=0, type=int)
    parser.add_argument('--base-lr', default=0.00001)
    parser.add_argument('--lr-decay-offset', default=80)
    parser.add_argument('--lr-decay-step', default=40)
    parser.add_argument('--lr-decay-rate', default=0.1)
    parser.add_argument('--momentum', default=0.9)
    parser.add_argument('--low-dim', default=62*256,
                        help='GaitSet model\'s output is (62,256)D, '
                             'then concated by 2 seq and gei features, last they need to be viewed')
    parser.add_argument('--npc-temperature', default=0.1, type=float)
    parser.add_argument('--npc-momentum', default=0.5, type=float)
    parser.add_argument('--load-pretrained', default=False,
                        help='decide whether load source domain\'s model parameters')
    cfg = parser.parse_args()

    return cfg