diff options
author | 2016-04-18 17:56:51 -0800 | |
---|---|---|
committer | 2016-04-18 19:03:29 -0700 | |
commit | 5c9bc51857bc0c330d3ab976871ee3509647d1e7 (patch) | |
tree | a58def7cbf316c6e091b3b36657f120f1388ec54 /tensorflow/models/embedding/word2vec.py | |
parent | fc432e37a7ddd408ff09a7b90b1c4cd5af1b134e (diff) |
Merge changes from github.
Change: 120185825
Diffstat (limited to 'tensorflow/models/embedding/word2vec.py')
-rw-r--r-- | tensorflow/models/embedding/word2vec.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/models/embedding/word2vec.py b/tensorflow/models/embedding/word2vec.py index cf30548e14..9cb15d3f41 100644 --- a/tensorflow/models/embedding/word2vec.py +++ b/tensorflow/models/embedding/word2vec.py @@ -82,15 +82,15 @@ flags.DEFINE_boolean( "interactive", False, "If true, enters an IPython interactive session to play with the trained " "model. E.g., try model.analogy('france', 'paris', 'russia') and " - "model.nearby(['proton', 'elephant', 'maxwell']") + "model.nearby(['proton', 'elephant', 'maxwell'])") flags.DEFINE_integer("statistics_interval", 5, "Print statistics every n seconds.") flags.DEFINE_integer("summary_interval", 5, "Save training summary to file every n seconds (rounded " - "up to statistics interval.") + "up to statistics interval).") flags.DEFINE_integer("checkpoint_interval", 600, "Checkpoint the model (i.e. save the parameters) every n " - "seconds (rounded up to statistics interval.") + "seconds (rounded up to statistics interval).") FLAGS = flags.FLAGS @@ -420,7 +420,7 @@ class Word2Vec(object): last_summary_time = now if now - last_checkpoint_time > opts.checkpoint_interval: self.saver.save(self._session, - opts.save_path + "model", + os.path.join(opts.save_path, "model.ckpt"), global_step=step.astype(int)) last_checkpoint_time = now if epoch != initial_epoch: |