Ejemplo n.º 1
0
def train_args(parser):
    parser = train_manager_args(parser)
    parser = preprocess_args(parser)
    expt_parser = parser.add_argument_group("Experiment arguments")
    expt_parser.add_argument('--expt-id', help='data file for training', default='')
    expt_parser.add_argument('--dataloader-type', help='Dataloader type.', choices=['normal', 'eeg', 'ml'], default='eeg')

    return parser
Ejemplo n.º 2
0
def train_args(parser):
    parser = train_manager_args(parser)
    parser = preprocess_args(parser)
    expt_parser = parser.add_argument_group("Experiment arguments")
    expt_parser.add_argument('--reproduce',
                             help='Method name for reproduction',
                             default='')
    expt_parser.add_argument('--expt-id',
                             help='data file for training',
                             default='')

    return parser
Ejemplo n.º 3
0
def train_args(parser):
    parser = train_manager_args(parser)
    parser = preprocess_args(parser)

    return parser