n11_train.py 文件源码

def add_fit_args(train):
    train.add_argument('--ngpus', default=1, type=int, help='amount of gpus')
    train.add_argument('--versn', default='rn-21', type=str, help='version of net')
    train.add_argument('--begin', default=0, type=int, help='start epoch')

    train.add_argument('--batch', default=8000, type=int, help='the batch size')
    train.add_argument('--nepoh', default=30, type=int, help='amount of epoch')
    train.add_argument('--check', default=20, type=int, help='period of check in iteration')
    train.add_argument('--lrate', default=0.001, type=float, help='start learning rate')
    train.add_argument('--optim', default='adam', type=str, help='optimizer')
    train.add_argument('--patin', default=15, type=int, help='waiting for n iteration without improvement')

    train.add_argument('--losss', default='categorical_crossentropy', type=str, help='loss function')
    train.add_argument('--mtype', default=1, type=int, help='neurons on branch audio')

    train.add_argument('--wpath', default=WPATH, type=str, help='net symbol path')
    train.add_argument('--dpath', default=FAST, type=str, help='data_path')
    train.add_argument('--split', default=200000, type=int, help='data_path')
    return train