def create_config(model_dir, hparams=None):
  """Creates config instance."""
  tf.logging.info("model_dir = " + model_dir)
  assert model_dir

  if hparams:
    tf.logging.info("Given override cfg:\n%s" % pprint.pformat(hparams))
  else:
    hparams = dict()

  # Build the default config
  cfg = model.build_config(model_dir=model_dir, data_path=FLAGS.data_path)

  if FLAGS.config_file:
    with tf.gfile.GFile(FLAGS.config_file) as f:
      file_cfg = json.load(f)
      file_cfg = utils.to_config(file_cfg)
    tf.logging.info("Loaded config from file:\n%s" % file_cfg)
    cfg = utils.merge_fixed_structure(cfg, file_cfg)

  # Override from flags
  overrides = dict()
  if FLAGS.config:
    overrides = utils.parse_config_string(FLAGS.config)
    tf.logging.info("Parsed config overrides:\n%s" % overrides)
    cfg = utils.merge_fixed_structure(cfg, overrides)

  if FLAGS.master:
    cfg.master = FLAGS.master

  cfg = utils.merge_fixed_structure(cfg, utils.unflatten_dict(hparams))

  tf.logging.info("Operative config:\n%s" % cfg)

  return cfg
示例#2
0
def config_page():
    if not request.method == "POST":
        return redirect(url_for("main"))
    else:
        config = utils.to_config(directory=session["TEMP_FOLDER"],
                                 config=request.form.to_dict())
        for file in os.listdir(session["TEMP_FOLDER"]):
            if file.endswith(".xlsx") or file.endswith(".csv"):
                file = os.path.join(session["TEMP_FOLDER"], file)
                break
        return redirect(
            url_for("analysis_page", survey_file=file, config=config))
def _load_config(model_dir):
  tf.logging.info("model_dir = " + model_dir)
  with tf.gfile.GFile(os.path.join(model_dir, "config.json")) as f:
    cfg = json.load(f)
    cfg = utils.to_config(cfg)
  return cfg