Exemplo n.º 1
0
def main():
    parser = argparse.ArgumentParser(description='Baseline')
    parser.add_argument('conf_path', type=str, metavar='conf_path',
                        help='input batch size for training (default: 64)')
    parser.add_argument('id', type=int, metavar='experiment_id',
                        help='Experiment ID')
    args = parser.parse_args()

    option = Option(args.conf_path)
    option.manualSeed = args.id + 1
    option.experiment_id = option.experiment_id + "{:0>2d}".format(args.id + 1)

    experiment = Experiment(option)
    experiment.pruning()
    experiment._load_resume()
    experiment.fine_tuning()
Exemplo n.º 2
0
Arquivo: main.py Projeto: qdmy/DCP-1
def main():
    parser = argparse.ArgumentParser(description='AuxNet')
    parser.add_argument(
        'conf_path',
        type=str,
        metavar='conf_path',
        help='the path of config file for training (default: 64)')
    parser.add_argument('id',
                        type=int,
                        metavar='experiment_id',
                        help='Experiment ID')
    args = parser.parse_args()

    option = Option(args.conf_path)
    option.manualSeed = args.id + 1
    option.experiment_id = option.experiment_id + "{:0>2d}".format(args.id + 1)

    experiment = Experiment(option)
    experiment.run()