Function main

code/cnn_1D_segm/train_fcn1D.py:346–377

()

def main():
    parser = argparse.ArgumentParser(description='FCN-1D model training')
    parser.add_argument('-dataset',
                        default='cortical_layers',
                        help='Dataset.')
    parser.add_argument('-learning_rate',
                        default=0.0005,
                        help='Learning Rate')
    parser.add_argument('--num_epochs',
                        '-ne',
                        type=int,
                        default=500,
                        help='Optional. Int to indicate the max'
                             'number of epochs.')
    parser.add_argument('-max_patience',
                        type=int,
                        default=25,
                        help='Max patience')
    parser.add_argument('-batch_size',
                        type=int,
                        nargs='+',
                        default=[1024, 1024, 1],
                        help='Batch size [train, val, test]. Default: -batch_size 1024 1024 1')
    parser.add_argument('-data_augmentation',
                        type=json.loads,
                        default={},
                        help='use data augmentation')
    args = parser.parse_args()

    train(dataset=args.dataset, learning_rate=args.learning_rate,
          num_epochs=args.num_epochs, max_patience=args.max_patience, data_augmentation=args.data_augmentation,
          batch_size=args.batch_size, savepath=SAVEPATH, loadpath=LOADPATH)


if __name__ == '__main__':