Loads environment for inference mode, used in jupyter notebook.
model_params = sketch_rnn_model.get_default_hparams()
with tf.gfile.Open(os.path.join(model_dir, "model_config.json"), "r") as f:
model_config = json.load(f)model_params.update(model_config)
return load_dataset(data_dir, model_params, inference_mode=True)
def load_model(model_dir):
After Change
Loads environment for inference mode, used in jupyter notebook.
model_params = sketch_rnn_model.get_default_hparams()
with tf.gfile.Open(os.path.join(model_dir, "model_config.json"), "r") as f:
model_params.parse_json(f.read())
return load_dataset(data_dir, model_params, inference_mode=True)
