Skip to content

Commit

Permalink
Update train.py
Browse files Browse the repository at this point in the history
  • Loading branch information
LargonD authored Jan 8, 2017
1 parent 2c61933 commit 16161c0
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@
tf.flags.DEFINE_integer("evaluate_every", 100, "Evaluate model on dev set after this many steps (default: 100)")
tf.flags.DEFINE_integer("checkpoint_every", 100, "Save model after this many steps (default: 100)")
tf.flags.DEFINE_integer("num_checkpoints", 5, "Number of checkpoints to store (default: 5)")
tf.flags.DEFINE_string("checkpoint_file", str(int(time.time())), "Checkpoint file for saving the training model (default: 'time.time()')")
# Misc Parameters
tf.flags.DEFINE_boolean("allow_soft_placement", True, "Allow device soft device placement")
tf.flags.DEFINE_boolean("log_device_placement", False, "Log placement of ops on devices")
Expand All @@ -43,7 +42,7 @@
print("")


# Data Preparatopn
# Data Preparation
# ==================================================

# Load data
Expand Down Expand Up @@ -105,7 +104,8 @@
grad_summaries_merged = tf.merge_summary(grad_summaries)

# Output directory for models and summaries
out_dir = os.path.abspath(os.path.join(os.path.curdir, "runs", FLAGS.checkpoint_file))
timestamp = str(int(time.time()))
out_dir = os.path.abspath(os.path.join(os.path.curdir, "runs", timestamp))
print("Writing to {}\n".format(out_dir))

# Summaries for loss and accuracy
Expand Down

0 comments on commit 16161c0

Please sign in to comment.