Skip to content

Commit

Permalink
Revert "timed checkpoint"
Browse files Browse the repository at this point in the history
This reverts commit f199be8.
  • Loading branch information
wballard committed May 16, 2016
1 parent f199be8 commit 58f3b04
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tflearn/models/dnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ class DNN(object):

def __init__(self, network, clip_gradients=5.0, tensorboard_verbose=0,
tensorboard_dir="/tmp/tflearn_logs/", checkpoint_path=None,
max_checkpoints=None, session=None, keep_checkpoint_every_n_hours=100.0 ):
max_checkpoints=None, session=None):
assert isinstance(network, tf.Tensor), "'network' arg is not a Tensor!"
self.net = network
self.train_ops = tf.get_collection(tf.GraphKeys.TRAIN_OPS)
Expand All @@ -54,7 +54,6 @@ def __init__(self, network, clip_gradients=5.0, tensorboard_verbose=0,
tensorboard_verbose=tensorboard_verbose,
checkpoint_path=checkpoint_path,
max_checkpoints=max_checkpoints,
keep_checkpoint_every_n_hours=keep_checkpoint_every_n_hours,
session=session)
self.session = self.trainer.session

Expand Down

0 comments on commit 58f3b04

Please sign in to comment.