摘要:
1.使用tf2onnx工具,把saved model转换为tf的graph def(不带function,也就是tf1的计算图) https://github.com/onnx/tensorflow-onnx/blob/v1.9.3/tf2onnx/tf_loader.py # -*- coding 阅读全文
摘要:
# -*- coding: utf-8 -*- from tensorflow.python.eager import def_function from tensorflow.python.framework import ops from tensorflow.python.keras.opti 阅读全文
摘要:
def get_layers_and_variables_from_model(model: tf.keras.Model, scope_name=None): layer_dict = {} if scope_name is not None: base_name = scope_name els 阅读全文
摘要:
1.从keras.Model获取output name from tensorflow.python.keras.saving import saving_utils function = saving_utils.trace_model_call(model) concrete_func = fu 阅读全文
摘要:
1.auc概念: ①随机挑选一对正负样本,正样本打分值大于负样本的概率 ②roc曲线面积 注意:按照定义①,这里有个问题,打分相同怎么取舍。这个问题可以根据roc曲线绘制方法可知,如果打分值相同的话,roc会有斜线产生,那么计算面积的时候就是梯形面积,等价于打分值相同的时候,取一半的正样本算作打分值 阅读全文
摘要:
std::string s = "1.3,1.4,1.5" 字符串中的数字按照逗号分割,把每个数字解析出来. void ParseStr(std::string& s, std::vector<float>& res) { char* pos = const_cast<char*>(s.c_str( 阅读全文
摘要:
环境:tensorflow2.2 使用tf.keras.Model.save保存saved_model格式时,默认的input和output比较通用,input_1, input2, output_1,output_2 自定义输入输出名字: import tensorflow as tf sigs 阅读全文
摘要:
gdb更友好的打印STL的内容,可以使用gcc提供的printers插件,只需将以下内容写到~/.gdbinit文件中即可: python import sys sys.path.insert(0, '/usr/share/gcc-7/python') from libstdcxx.v6.print 阅读全文
摘要:
Tensorflow1.13.1中关于nccl的源码位于:tensorflow\core\kernels\nccl_ops.cc;tensorflow\core\nccl\nccl_manager.cc其中核心代码位于nccl_manager.cc文件,入口API为:AddToAllReduce 想 阅读全文
摘要:
在~/.bashrc文件中加入如下内容: alias ll="ls -lh " alias gs="git status " alias gc="git checkout " alias gb="git branch " alias gm="git merge " parse_git_branch( 阅读全文