NN格式
weixin_39087379
这个作者很懒,什么都没留下…
展开
-
获取ckpt中的变量,并打印变量值
with graph.as_default(): reader = tf.pywrap_tensorflow.NewCheckpointReader(ckpt_path) var_to_shape_map = reader.get_variable_to_shape_map() # 获取变量名 for key in var_to_shape_map: print("tensor_name: ", key) value = reader.get_ten原创 2021-10-22 17:37:18 · 183 阅读 · 0 评论 -
tf1加载tf2权重流程:
tf1加载tf2权重流程:1、获取到变量:var=get_variables(var_name) 或者tf.get_collection_ref(tf.GraphKeys.GLOBAL_VARIABLES)]2、变量赋值 : a=tf.assign(var,value) 或者 a=var.assign(value)3、变量赋值生效: sess.run(a)原创 2021-10-22 17:26:15 · 227 阅读 · 0 评论 -
tensorflow打印checkpoint里的变量名称和值
reader = tf.pywrap_tensorflow.NewCheckpointReader(ckpt_path)var_to_shape_map = reader.get_variable_to_shape_map()for key in var_to_shape_map:print("tensor_name: ", key)value=reader.get_tensor(key)if isinstance(value, np.ndarray):print(‘value:’,np.sha原创 2021-10-21 21:55:00 · 197 阅读 · 0 评论 -
注意力机制综述:基础原理、变种和最近研究
转载:https://www.pianshen.com/article/90361180727/原创 2021-08-18 20:57:35 · 69 阅读 · 0 评论 -
conformer
待续原创 2021-08-03 15:17:18 · 92 阅读 · 0 评论 -
TFLITE-1-模型转换
test原创 2021-03-03 15:22:39 · 126 阅读 · 1 评论 -
植入tflite源码,需要改动的内容
植入tflite源码,需要改动的内容: 1、修改interpreter.h , 改结构体Subgraph 为需要替换的名字, 删除声明TfLiteStatus Invoke(); 增加新的定义,为防止调用出错,将函数名改为TfLiteStatus Invoke_x() {},代码为interpreter中相应函数代码的移植 2、修改 将sugbraph.h 添加subgrap.h移动到工程目录,并重命名为 sugbraphx.h 修改结构体名称 Subgrap 为 Subgrapx, 可原创 2021-03-01 10:06:49 · 204 阅读 · 0 评论