self.saver = tf.train.Saver(tf.trainable_variables() + tf.get_collection_ref('bn_collections'))
总是出错,只好注释掉这句话,不保存变量了
但是不可行,后面会调用保存的模型,所以改成
self.saver = tf.train.Saver(tf.get_collection_ref('bn_collections'))
self.saver = tf.train.Saver(tf.trainable_variables() + tf.get_collection_ref('bn_collections'))
总是出错,只好注释掉这句话,不保存变量了
但是不可行,后面会调用保存的模型,所以改成
self.saver = tf.train.Saver(tf.get_collection_ref('bn_collections'))