python - 使用 TensorFlow 2 在训练期间记录权重规范
问题描述
我正在使用 tensorflow 训练一个神经网络,我想监控一些层的 L2 范数。我在 tensorflow 1 中找到了如何做到这一点,例如在这里,但我找不到在 tensorflow 2 中做同样事情的方法。我想知道我该怎么做。谢谢!
解决方案
Tensorflow 2 提供了 Tensorflow 1 的兼容版本。请找到以下用于计算非零权重的代码
import tensorflow as tf
import numpy as np
#Replace tf.trainable_variables() with tf.compat.v1.trainable_variables
tvars = tf.compat.v1.trainable_variables()
nonzero_parameters = np.sum([np.count_nonzero(var) for var in tvars])
推荐阅读
- php - 在我的 API 中实现 elasticsearch-php 客户端的建议功能
- jquery - 如何在codeigniter示例中使用foreach($data as $buff)进行双循环首先选择目的地,第二种形式选择是出发
- python - Pyhon3.4 TypeError:'dict_keys'对象不支持索引
- r - 使用 purrr 中的 map() 绘制时间序列
- android - 如何设置随机渐变背景颜色?
- php - 如果我们有数组索引的概念,如何使用 array_column 函数从数组中获取记录?
- division - 如何有效地找到整数 k 的值,其中 q 有限地除 b ^ k?
- c# - 如何等到 C# 中的方法完成?
- asp.net - ASP.Net UserControls 不触发任何 AutoPost Back 或 Click 事件
- python - 在熊猫中选择包含关键字的列名称