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)
