Ejemplos de set_up_logging en Python

Ejemplo n.º 1

Archivo: train.py Proyecto: chychen/RL_strategies

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))

Ejemplo n.º 2

Archivo: train.py Proyecto: chychen/RL_strategies

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)

Ejemplo n.º 3

def main(_):
    """Load a trained algorithm and render videos."""
    utility.set_up_logging()
    if not FLAGS.logdir or not FLAGS.outdir:
        raise KeyError('You must specify logging and outdirs directories.')
    FLAGS.logdir = os.path.expanduser(FLAGS.logdir)
    FLAGS.outdir = os.path.expanduser(FLAGS.outdir)
    visualize(FLAGS.logdir, FLAGS.outdir, FLAGS.num_agents, FLAGS.num_episodes,
              FLAGS.checkpoint, FLAGS.env_processes)

Ejemplo n.º 4

Archivo: visualize.py Proyecto: shamanez/agents

def main(_):
  """Load a trained algorithm and render videos."""
  utility.set_up_logging()
  if not FLAGS.logdir or not FLAGS.outdir:
    raise KeyError('You must specify logging and outdirs directories.')
  FLAGS.logdir = os.path.expanduser(FLAGS.logdir)
  FLAGS.outdir = os.path.expanduser(FLAGS.outdir)
  visualize(
      FLAGS.logdir, FLAGS.outdir, FLAGS.num_agents, FLAGS.num_episodes,
      FLAGS.checkpoint, FLAGS.env_processes)

Ejemplo n.º 5

Archivo: train_ppo.py Proyecto: Valentactive/bullet3

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))

Ejemplo n.º 6

Archivo: visualize.py Proyecto: Cynthia1123/Chess-Einstein

def main(_):
    FLAGS.logdir = '../../Log/20170920T160752-pendulum'
    FLAGS.outdir = '../../Log/20170920T160752-pendulum_out'
    FLAGS.num_agents = 1
    FLAGS.num_episodes = 5
    FLAGS.env_processes = False

    """Load a trained algorithm and render videos."""
    utility.set_up_logging()
    if not FLAGS.logdir or not FLAGS.outdir:
        raise KeyError('You must specify logging and outdirs directories.')
    FLAGS.logdir = os.path.expanduser(FLAGS.logdir)
    FLAGS.outdir = os.path.expanduser(FLAGS.outdir)
    visualize(
        FLAGS.logdir, FLAGS.outdir, FLAGS.num_agents, FLAGS.num_episodes,
        FLAGS.checkpoint, FLAGS.env_processes)

Ejemplo n.º 7

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))

Ejemplo n.º 8

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)

Ejemplo n.º 9

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)