Tensorflow
文章平均质量分 75
楓尘林间
计算机爱好者
展开
-
tensorflow2.0 的Layer 的 call build init方法
官方API显然,这三个函数都是从tf.keras.layers.Layer处继承而来的。_init_ 可以在其中进行所有与输入无关的初始化build 知道输入张量的形状,并可以进行其余的初始化call 可以在其中进行前向计算官方API的例子:class MyDenseLayer(tf.keras.layers.Layer): def __init__(self, num_outputs): super(MyDenseLayer, self).__init__() sel原创 2021-05-12 16:31:04 · 1368 阅读 · 0 评论 -
tensorflow中如何判断两个tensor值相等(tensor转array)
需要转化为numpy进行判断转化为numpy的方法有两种方法1: tf.Session().run(tensorA) 转化为numpyprint(type(tf.Session().run(tf.constant([1,2]))))结果:<class ‘numpy.ndarray’>方法2: 使用eval()方法sess = tf.InteractiveSession()print(type(tf.constant([1,3]).eval()))或sess = tf.原创 2020-12-03 14:49:11 · 1765 阅读 · 6 评论 -
tensorflow利用saver读取部分参数变量值
做实验时到的问题:'feature_embeddings:0' not found in checkpoint当时的实验室分别训练modelA和modelB,再将B模型的参数载入到A中,具体如下图所示:modelA中包含modelB中所有参数,可以将modelA中参数载入ModelB,但是反过来则报错,具体载入语句为:# 存储saver = tf.train.Saver(max_to_keep=5) saver.save(self.sess, self.save_path + 'model.原创 2020-10-21 16:57:06 · 450 阅读 · 0 评论 -
tensorflow ---如何只更新部分参数
trainable_vars = tf.trainable_variables()freeze_conv_var_list = [t for t in trainable_vars if not t.name.startswith(u'conv')]grads = opt.compute_gradients(loss, var_list=freeze_conv_var_list)参考:https://www.cnblogs.com/hrlnw/p/10400057.html在优化时,默认是计算更新原创 2020-11-12 11:15:26 · 973 阅读 · 0 评论 -
tensorflow2.0安装 CUDA和CUDNN并运行成功
查看 CUDA 版本:cat /usr/local/cuda/version.txt或者:nvcc -V查看 CUDNN 版本:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看能否使用gpu:jupyter输入:import tensorflow as tfprint("Num GPUs Available: ", len(tf.config.experimental.list_physical_devi原创 2020-09-06 19:57:00 · 1743 阅读 · 1 评论 -
tensorflow2.0利用keras 打印AUC指标
keras中没有对AUC的定义,需要我们自己定义:x = np.linspace(-6, 6, 200)y = np.array([0.0]*100 + [1.0]*100)state = np.random.get_state()np.random.shuffle(x)np.random.set_state(state)np.random.shuffle(y)x_train, y_train = x[0:160], y[0:160]plt.scatter(x_train, y_trai原创 2020-09-03 16:07:25 · 2694 阅读 · 0 评论