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)