灰太狼锅锅

摘要： 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( 阅读全文