tensorflow2 使用BN Dropout 时的开关

最新推荐文章于 2024-03-23 06:00:00 发布

coolsunxu

最新推荐文章于 2024-03-23 06:00:00 发布

阅读量858

点赞数 3

分类专栏： Tensorflow Python 文章标签： tensorflow2 BN Dropout

本文链接：https://blog.csdn.net/coolsunxu/article/details/104171781

版权

Python 同时被 2 个专栏收录

55 篇文章 2 订阅

订阅专栏

Tensorflow

19 篇文章 0 订阅

订阅专栏


import  tensorflow as tf
from tensorflow.keras import layers

# 训练阶段
with tf.GradientTape() as tape:
	
	out = network(x, training=True) # 类似 pytorch 的 model.train()

# 在测试阶段，需要设置 training=False， 避免 BN 层采用错误的行为：
for x,y in db_test: # 遍历测试集
	
	out = network(x, training=False) # 类似 pytorch 的 model.eval()

# 上述同样适用于 dropout