(self, sess=None)
| 74 | print("Model saved in file: %s" % save_path) |
| 75 | |
| 76 | def load(self, sess=None): |
| 77 | if not sess: |
| 78 | raise AttributeError("TensorFlow session not provided.") |
| 79 | saver = tf.train.Saver(self.vars) |
| 80 | save_path = "tmp/%s.ckpt" % self.name |
| 81 | saver.restore(sess, save_path) |
| 82 | print("Model restored from file: %s" % save_path) |
| 83 | |
| 84 | |
| 85 | class MLP(Model): |