Skip to content
This repository was archived by the owner on Aug 31, 2021. It is now read-only.

Commit 7cdfb0d

Browse files
committed
Added continue_training=True to the estimator in resnet to make sure it keeps training
1 parent d0ac8de commit 7cdfb0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/resnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def res_net(x, y, activation=tf.nn.relu):
140140
# Create a new resnet classifier.
141141
classifier = skflow.TensorFlowEstimator(
142142
model_fn=res_net, n_classes=10, batch_size=100, steps=100,
143-
learning_rate=0.001)
143+
learning_rate=0.001, continue_training=True)
144144

145145
while True:
146146
# Train model and save summaries into logdir.

0 commit comments

Comments
 (0)