Ejemplo n.º 1
0
def main():
    parser = build_parser()
    args = vars(parser.parse_args())
    with open('ppo.json') as f:
        config = json.load(f)
        if args['mode'] == 'download':
            from data.download_data import DataDownloader
            data_downloader = DataDownloader(config)
            data_downloader.save_data()
        else:
            session(config, args)
def main():
    logger.info("Initializing")
    args = parser_args()
    tf.compat.v1.set_random_seed(args.seed)
    random.seed(args.seed)
    with open(args.config) as f:
        config = json.load(f)
        if args.mode == 'download':
            from data.download_data import DataDownloader
            data_downloader = DataDownloader(config)
            data_downloader.save_data()
        else:
            session(config, args)