错误记录
2020-05-14 10:11:00.935025: E tensorflow/stream_executor/cuda/cuda_dnn.cc:385] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
2020-05-14 10:11:00.935050: E tensorflow/stream_executor/cuda/cuda_dnn.cc:352] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM
2020-05-14 10:11:00.935055: F tensorflow/core/kernels/conv_ops.cc:717] Check failed: stream->parent()->GetConvolveAlgorithms( conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms)
一种情况剖析
config = tf.ConfigProto(allow_soft_placement=True)
sess = tf.Session(config=config)
net = get_network("VGGnet_test")
saver = tf.train.Saver()
saver.restore(sess, os.path.join(os.getcwd(),"models/VGGnet_fast_rcnn_iter_470000.ckpt"))
print (' done.')
def load_tf_model():
config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.allow_growth = True ###
config.gpu_options.per_process_gpu_memory_fraction = 0.33 ###
sess = tf.Session(config=config)
net = get_network("VGGnet_test")
saver = tf.train.Saver()
ckpt = tf.train.get_checkpoint_state('models/')
print(ckpt.model_checkpoint_path)
saver.restore(sess, os.path.join(os.getcwd(),ckpt.model_checkpoint_path))
return sess,saver,net
ocr_ctpn_sess,ocr_ctpn_saver,ocr_ctpn_net = load_tf_model()
ocr_ctpn_sess,ocr_ctpn_saver,ocr_ctpn_net作为全局变量传递需要注意下
恢复点加载
reader = pywrap_tensorflow.NewCheckpointReader(pretrained_model)
var_to_shape_map = reader.get_variable_to_shape_map()
base_weights = reader.get_tensor('model/mel_encoder/mel_encoder/spk_logits/kernel')
base_bias = reader.get_tensor('model/mel_encoder/mel_encoder/spk_logits/bias')
base_weights = sess.graph.get_tensor_by_name('model/optimizer/model/mel_encoder/mel_encoder/spk_logits/kernel/Adam_1:0')