阿门的博客

# A logit (also called a score) is a raw unscaled value associated with a class.# 多分类问题 输入标签需要Onehot编码 输出logits向量tf.nn.softmax_cross_entropy_with_logits# 多分类问题 输入标签不必Onehot编码 输出logits向量tf.nn.spar...

计算模型需要训练的参数数目：def count_trainable_vars(): total_parameters = 0 for variable in tf.trainable_variables(): variable_parameters = 1 for dim in variable.get_shape(): ...

一：用xx.npz文件初始化网络使用tensorpack框架的时候，发现官方提供的训练好的权重文件是xx.npz格式的，我想将其某些层的参数用在自己的网络中。import osimport randomimport tensorflow as tfimport numpy as npPRE_IMANET_NPZ = 'XX.npz'def convert_param_n...

问题描述：加载部分预训练好的权重到自己的模型上，fine-tune网络的时候，希望已经具有预训练权重的部分学习速率小一些，随机初始化的新添加的层学习速率大一些。方法：用 apply_gradients()函数。代码：import tensorflow as tf# the variables waiting for optimizationx = tf.Variabl...

搜索安装包并选择合适版本：# 搜索当前可用的安装包版本anaconda search -t conda tensorflow# 选择合适版本，查询安装命令anaconda show anaconda/tensorflow-gpu 注意tensorflow的版本要与CUDA匹配，我是CUDA10，下载了tensorflow1.12结果：按照最后一行所示命令进行安装...