阿门的博客

# 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 os import random import tensorflow as tf import numpy as np PRE_IMANET_NPZ = 'XX.npz' def convert_param_n...

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

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