
tf
siqi_fighting
这个作者很懒,什么都没留下…
展开
-
tensorflow统计graph中的trainable_variables
最简单的做法: 转自: https://blog.csdn.net/feynman233/article/details/79187304, 版权归原作者所有。 print(np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])) 另有篇博客讲解的很详细:原文地址https://blog.csd...转载 2018-07-19 20:07:17 · 2095 阅读 · 0 评论 -
tensorflow 仅初始化指定的variables及未初始化的variables
1. 已知variables v0,v1,v2,对特定的variables进行初始化可使用以下方法: initialize_op = tf.variables_initializer([v0,v1,v2]) sess.run(initialize_op) 2. restore graph 进行了fine_tuning, 但仍保留restored weights,仅对新的未被初始化的tens...原创 2018-07-20 14:46:05 · 2124 阅读 · 0 评论 -
tensorflow 获取graph中的所有tensor
[n.name for n in tf.get_default_graph().as_graph_def().node] 摘自https://stackoverflow.com/questions/36883949/in-tensorflow-get-the-names-of-all-the-tensors-in-a-graph原创 2018-07-20 15:44:24 · 17081 阅读 · 1 评论