def run_tpu(no_resign=False): os.environ[ 'GRPC_DEFAULT_SSL_ROOTS_FILE_PATH'] = '/etc/ssl/certs/ca-certificates.crt' flagset = [ 'bazel-bin/cc/main', '--mode=selfplay', '--engine=tpu', '--model={}'.format( os.path.join(fsdb.working_dir(), 'model.ckpt-%d.pb')), '--output_dir={}'.format(fsdb.selfplay_dir()), '--holdout_dir={}'.format(fsdb.holdout_dir()), '--sgf_dir={}'.format(fsdb.sgf_dir()), '--run_forever=true', '--output_bigtable={}'.format(FLAGS.output_bigtable) ] if 'KUBE_GOOGLE_CLOUD_TPU_ENDPOINTS' in os.environ: flagset.append('--tpu_name={}'.format( os.environ['KUBE_GOOGLE_CLOUD_TPU_ENDPOINTS'])) if no_resign: flagset.extend(['--flagfile=rl_loop/distributed_flags_nr']) else: flagset.extend([ '--flags_path={}'.format(fsdb.flags_path()), '--flagfile=rl_loop/distributed_flags' ]) mask_flags.checked_run(flagset)
def run_tpu(): mask_flags.checked_run([ 'bazel-bin/cc/main', '--mode=selfplay', '--engine=tpu', '--checkpoint_dir={}'.format(fsdb.working_dir()), '--output_dir={}'.format(fsdb.selfplay_dir()), '--holdout_dir={}'.format(fsdb.holdout_dir()), '--sgf_dir={}'.format( fsdb.sgf_dir()), '--flags_path={}'.format(fsdb.flags_path()), '--run_forever=true', '--flagfile=rl_loop/distributed_flags' ])
def main(argv): if len(argv) > 1: raise app.UsageError('Too many command-line arguments.') if not 'CBT_TABLE' in os.environ: raise app.UsageError('CBT_TABLE not set') while True: new_pct = get_95_percentile_bleak() update_flagfile(fsdb.flags_path(), new_pct) time.sleep(60 * 3)
def main(argv): if len(argv) > 1: raise app.UsageError('Too many command-line arguments.') games_nr = bigtable_input.GameQueue(FLAGS.cbt_project, FLAGS.cbt_instance, FLAGS.cbt_table + '-nr') while True: new_pct = get_95_percentile_bleak(games_nr) update_flagfile(fsdb.flags_path(), new_pct) time.sleep(60 * 3)