重慶分公司,新征程啟航
為企業(yè)提供網(wǎng)站建設(shè)、域名注冊、服務(wù)器等服務(wù)
為企業(yè)提供網(wǎng)站建設(shè)、域名注冊、服務(wù)器等服務(wù)
終于找到bug原因!記一下;還是不熟悉平臺的原因造成的!
Q:為什么會出現(xiàn)兩個模型對象在同一個文件中一起運行,當(dāng)直接讀取他們分開運行時訓(xùn)練出來的模型會出錯,而且總是有一個正確,一個讀取錯誤? 而 直接在同一個文件又訓(xùn)練又重新加載模型預(yù)測不出錯,而且更詭異的是此時用分文件里的對象加載模型不會出錯?
model.py,里面含有 ModelV 和 ModelP,另外還有 modelP.py 和 modelV.py 分別只含有 ModelP 和 ModeV 這兩個對象,先使用 modelP.py 和 modelV.py 分別訓(xùn)練好模型,然后再在 model.py 里加載進(jìn)來:
# -*- coding: utf8 -*- import tensorflow as tf class ModelV(): def __init__(self): self.v1 = tf.Variable(66, name="v1") self.v2 = tf.Variable(77, name="v2") self.save_path = "model_v/model.ckpt" self.init = tf.global_variables_initializer() self.saver = tf.train.Saver() self.sess = tf.Session() def train(self): self.sess.run(self.init) print 'v2', self.v2.eval(self.sess) self.saver.save(self.sess, self.save_path) print "ModelV saved." def predict(self): all_vars = tf.trainable_variables() for v in all_vars: print(v.name) self.saver.restore(self.sess, self.save_path) print "ModelV restored." print 'v2', self.v2.eval(self.sess) print '------------------------------------------------------------------' class ModelP(): def __init__(self): self.p1 = tf.Variable(88, name="p1") self.p2 = tf.Variable(99, name="p2") self.save_path = "model_p/model.ckpt" self.init = tf.global_variables_initializer() self.saver = tf.train.Saver() self.sess = tf.Session() def train(self): self.sess.run(self.init) print 'p2', self.p2.eval(self.sess) self.saver.save(self.sess, self.save_path) print "ModelP saved." def predict(self): all_vars = tf.trainable_variables() for v in all_vars: print v.name self.saver.restore(self.sess, self.save_path) print "ModelP restored." print 'p2', self.p2.eval(self.sess) print '---------------------------------------------------------------------' if __name__ == '__main__': v = ModelV() p = ModelP() v.predict() #v.train() p.predict() #p.train()