def main(_): """ Create or load configuration and launch the trainer. """ utility.set_up_logging() if not FLAGS.resume: logdir = FLAGS.logdir and os.path.expanduser( os.path.join(FLAGS.logdir, '{}-{}'.format(FLAGS.timestamp, FLAGS.config))) else: logdir = FLAGS.logdir if FLAGS.vis: outdir = os.path.join(logdir, 'train_output') else: outdir = None try: config = utility.load_config(logdir) except IOError: if not FLAGS.config: raise KeyError('You must specify a configuration.') config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) for score in train(config, FLAGS.env_processes, outdir): tf.logging.info('Score {}.'.format(score))
def main(_): """ Create or load configuration and launch the trainer. """ if FLAGS.config == 'offense': data = np.load('bball_strategies/pretrain/data/off_obs.npy') label = np.load('bball_strategies/pretrain/data/off_actions.npy') elif FLAGS.config == 'defense': data = np.load('bball_strategies/pretrain/data/def_obs.npy') label = np.load('bball_strategies/pretrain/data/def_actions.npy') else: raise ValueError('{} is not an available config'.format(FLAGS.config)) utility.set_up_logging() if not FLAGS.resume: logdir = FLAGS.logdir and os.path.expanduser( os.path.join(FLAGS.logdir, '{}-{}'.format(FLAGS.timestamp, FLAGS.config))) else: logdir = FLAGS.logdir if FLAGS.vis: outdir = os.path.join(logdir, 'train_output') else: outdir = None try: config = utility.load_config(logdir) except IOError: if not FLAGS.config: raise KeyError('You must specify a configuration.') config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) train(config, data, label, outdir)
def main(_): """Create or load configuration and launch the trainer.""" utility.set_up_logging() if not FLAGS.config: raise KeyError('You must specify a configuration.') logdir = FLAGS.logdir and os.path.expanduser(os.path.join( FLAGS.logdir, '{}-{}'.format(FLAGS.timestamp, FLAGS.config))) try: config = utility.load_config(logdir) except IOError: config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) for score in train(config, FLAGS.env_processes): tf.logging.info('Score {}.'.format(score))
def main(_): FLAGS.logdir = '../../Log' FLAGS.config = 'pendulum' FLAGS.env_processes = False """Create or load configuration and launch the trainer.""" utility.set_up_logging() if not FLAGS.config: raise KeyError('You must specify a configuration.') logdir = FLAGS.logdir and os.path.expanduser( os.path.join(FLAGS.logdir, '{}-{}'.format(FLAGS.timestamp, FLAGS.config))) try: config = utility.load_config(logdir) except IOError: config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) global globalConfig globalConfig = config for score in train(config, FLAGS.env_processes): tf.logging.info('Score {}.'.format(score))
def main(_): """ Create or load configuration and launch the trainer. """ utility.set_up_logging() if FLAGS.resume: logdir = FLAGS.logdir else: logdir = FLAGS.logdir and os.path.expanduser(os.path.join( FLAGS.logdir, '{}-{}'.format(FLAGS.timestamp, FLAGS.config))) if FLAGS.vis: outdir = os.path.join(logdir, 'train_output') else: outdir = None if not FLAGS.config: raise KeyError('You must specify a configuration.') config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) # collecting testing(config, FLAGS.env_processes, outdir)
def main(_): """ Create or load configuration and launch the trainer. """ off_data = np.load('bball_strategies/pretrain/data/off_obs.npy') off_label = np.load('bball_strategies/pretrain/data/off_actions.npy') def_data = np.load('bball_strategies/pretrain/data/def_obs.npy') def_label = np.load('bball_strategies/pretrain/data/def_actions.npy') utility.set_up_logging() logdir = FLAGS.logdir try: config = utility.load_config(logdir) except IOError: if not FLAGS.config: raise KeyError('You must specify a configuration.') config = tools.AttrDict(getattr(configs, FLAGS.config)()) config = utility.save_config(config, logdir) outdir = os.path.expanduser(os.path.join(FLAGS.logdir, 'vis')) vis_data(off_data, off_label, def_data, def_label, outdir, start_idx=0) testing(config, off_data, off_label, def_data, def_label, outdir)