def init(self):
if self.load_path:
print 'Attempting to load directly from path:',
print self.load_path
self.saver.restore(self.sess,self.load_path)
else:
print 'New ENCODE Model..init new Z parameters'
init=tf.variables_initializer(var_list=self.var)
print 'Initializing following variables:'
for v in self.var:
print v.name, v.get_shape().as_list()
self.model.sess.run(init)
评论列表
文章目录