(self, sess=None)
| 93 | raise NotImplementedError |
| 94 | |
| 95 | def save(self, sess=None): |
| 96 | if not sess: |
| 97 | raise AttributeError("TensorFlow session not provided.") |
| 98 | saver = tf.train.Saver(self.vars) |
| 99 | save_path = saver.save(sess, "Data/checkpoint/%s.ckpt" % self.name) |
| 100 | print("Model saved in file: %s" % save_path) |
| 101 | |
| 102 | def load(self, sess=None): |
| 103 | if not sess: |