Exemplo n.º 1
0
        'hidden_auxNnode': ['20'],
        'batch_size': ['10000', '15000'],
        'problem': ['1'],
        'n_iteraction': ['100'],
        'epochs': ['1', '3'],
        'alr': ['0.0000001'],
        'amomentum': ['0.6'],
        'lam': ['0.1'],
    }

    if jobname == 'DNN':
        job_array = DNN_job_array
    elif jobname == 'ANN' or jobname == 'ANNReg':
        job_array = ANN_job_array
    else:
        raise RuntimeError('Unknown job name: {}'.format(jobname))


    if args.command == '_run':
        batch._run(args._run)

    elif args.command == 'htc':
        batch.create_jdl(job_array)
        batch.create_wrap()

    elif args.command == 'all':
        batch.create_jdl(job_array, local_run = True)

    else:
        raise RuntimeError('Unknown command: {}'.format(args.command))