import argparse

def get_args_parser():

    parser = argparse.ArgumentParser('EEGTransformer training and evaluation script', add_help=False)

    parser.add_argument('--opt', default='adamw', type=str)

    parser.add_argument('--epochs', default=50, type=int)

#     parser.add_argument('--lr', default=1e-5, type=float)

#     parser.add_argument('--sched', default='cosine', type=str)

    parser.add_argument('--momentum', type=float, default=0.9, metavar='M',

                        help='SGD momentum (default: 0.9)')

    parser.add_argument('--weight-decay', type=float, default=0.05,

                        help='weight decay (default: 0.05)')

#     parser.add_argument('--min-lr', type=float, default=1e-5, metavar='LR',

#                         help='lower lr bound for cyclic schedulers that hit 0 (1e-5)')

        # Learning rate schedule parameters

    parser.add_argument('--sched', default='cosine', type=str, metavar='SCHEDULER',

                        help='LR scheduler (default: "cosine"')

    parser.add_argument('--lr', type=float, default=5e-6, metavar='LR',

                        help='learning rate (default: 5e-4)')

    parser.add_argument('--lr-noise', type=float, nargs='+', default=None, metavar='pct, pct',

                        help='learning rate noise on/off epoch percentages')

    parser.add_argument('--lr-noise-pct', type=float, default=0.67, metavar='PERCENT',

                        help='learning rate noise limit percent (default: 0.67)')

    parser.add_argument('--lr-noise-std', type=float, default=1.0, metavar='STDDEV',

                        help='learning rate noise std-dev (default: 1.0)')

    parser.add_argument('--warmup-lr', type=float, default=1e-6, metavar='LR',

                        help='warmup learning rate (default: 1e-6)')

    parser.add_argument('--min-lr', type=float, default=1e-5, metavar='LR',

                        help='lower lr bound for cyclic schedulers that hit 0 (1e-5)')

    parser.add_argument('--hard_code_lr', type=float, default=0.0, metavar='LR',

                        help='learning rate (default: 5e-4)')

    parser.add_argument('--decay-epochs', type=float, default=30, metavar='N',

                        help='epoch interval to decay LR')

    parser.add_argument('--warmup-epochs', type=int, default=0, metavar='N',

                        help='epochs to warmup LR, if scheduler supports')

    parser.add_argument('--cooldown-epochs', type=int, default=10, metavar='N',

                        help='epochs to cooldown LR at min_lr, after cyclic schedule ends')

    parser.add_argument('--patience-epochs', type=int, default=10, metavar='N',

                        help='patience epochs for Plateau LR scheduler (default: 10')

    parser.add_argument('--decay-rate', '--dr', type=float, default=0.1, metavar='RATE',

                        help='LR decay rate (default: 0.1)')

    return parser