Exports tensorflow model
(self,folder)
| 86 | self.load(self.tmp_model) |
| 87 | |
| 88 | def save(self,folder): |
| 89 | """Exports tensorflow model |
| 90 | """ |
| 91 | tf.add_to_collection('tf_input',self.tf_input) |
| 92 | tf.add_to_collection('tf_output',self.tf_output) |
| 93 | for var_name,var in self.tf_constants_vars: |
| 94 | tf.add_to_collection(var_name,var) |
| 95 | saver = tf.train.Saver() |
| 96 | saver.save(self.sess,'%s/tensorflow-model'%folder) |
| 97 | |
| 98 | def load(self,folder): |
| 99 | self.sess = tf.Session() |