Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.Sign up
.Save, .Restore for contrib.learn.DNNClassifier #3306
The older functions like skflow.TensorFlowDNNClassifier had methods .save and .restore. These were supposedly migrated over to the contrib.learn functions, but there are no longer save and restore methods that I can find. Is there a new protocol for saving the graph and variables with the new contrib.learn package?
Checkpoints are saved automatically, and if you specify the same directory when you create another Estimator (of the same kind/same parameters), it will load the checkpoints and you can continue training.
There is currently a bug which prevents you from using the restored Estimator for inference without running at least one step of training.
I will close this issue -- for detailed questions, please ask on StackOverflow, we can track questions and answers better over there.