Python add_dataset_flagsの例

コード例 #1

ファイル: train.py プロジェクト: PeterouZh/MEAL-V2

def parse_args(argv):
    """Parse arguments @argv and return the flags needed for training."""
    parser = argparse.ArgumentParser(description=__doc__, allow_abbrev=False)

    group = parser.add_argument_group('General Options')
    opts.add_general_flags(group)

    group = parser.add_argument_group('Dataset Options')
    opts.add_dataset_flags(group)

    group = parser.add_argument_group('Model Options')
    opts.add_model_flags(group)

    group = parser.add_argument_group('Soft Label Options')
    opts.add_teacher_flags(group)

    group = parser.add_argument_group('Training Options')
    opts.add_training_flags(group)

    update_parser_defaults_from_yaml(parser)
    args = parser.parse_args(argv)

    # if args.teacher_model is not None and args.teacher_state_file is None:
    #     parser.error("You should set --teacher-state-file if "
    #                  "--teacher-model is set.")

    return args

コード例 #2

ファイル: train_sp.py プロジェクト: logichen/UCI-CIFAR-100

def parse_args(argv):
    """Parse arguments @argv and return the flags needed for training."""
    parser = argparse.ArgumentParser(description=__doc__, allow_abbrev=False)

    group = parser.add_argument_group('General Options')
    opts.add_general_flags(group)

    group = parser.add_argument_group('Dataset Options')
    opts.add_dataset_flags(group)

    group = parser.add_argument_group('Model Options')
    opts.add_model_flags(group)

    group = parser.add_argument_group('Label Refinery Options')
    opts.add_label_refinery_flags(group)

    group = parser.add_argument_group('Training Options')
    opts.add_training_flags(group)

    args = parser.parse_args(argv)

    if args.label_refinery_model is not None and args.label_refinery_state_file is None:
        parser.error("You should set --label-refinery-state-file if "
                     "--label-refinery-model is set.")

    return args

コード例 #3

ファイル: FKD_train.py プロジェクト: szq0214/MEAL-V2

def parse_args(argv):
    """Parse arguments @argv and return the flags needed for training."""
    parser = argparse.ArgumentParser(description=__doc__, allow_abbrev=False)

    group = parser.add_argument_group('General Options')
    opts.add_general_flags(group)

    group = parser.add_argument_group('Dataset Options')
    opts.add_dataset_flags(group)

    group = parser.add_argument_group('Model Options')
    opts.add_model_flags(group)

    group = parser.add_argument_group('Soft Label Options')
    opts.add_teacher_flags(group)

    group = parser.add_argument_group('Training Options')
    opts.add_training_flags(group)

    group = parser.add_argument_group('CutMix Training Options')
    opts.add_cutmix_training_flags(group)

    args = parser.parse_args(argv)

    return args

コード例 #4

def parse_args(argv):
    """Parse arguments @argv and return the flags needed for training."""
    parser = argparse.ArgumentParser(description=__doc__, allow_abbrev=False)

    group = parser.add_argument_group('General Options')
    opts.add_general_flags(group)

    group = parser.add_argument_group('Dataset Options')
    opts.add_dataset_flags(group)

    group = parser.add_argument_group('Model Options')
    opts.add_model_flags(group)

    args = parser.parse_args(argv)

    if args.model_state_file is None:
        parser.error("You should set --model-state-file to reload a model "
                     "state.")

    return args