tensflow2.0系列-反向训练参数问题
在2.0中进行TNN训练反向设置参数时,由于是自定义训练步骤,且每步输入的batch_size = 1 的数据时,在参数设计神经网络中不出现tf.keras.layers.BatchNormalization()时训练效果会更好,如下图所示
655 字
|
3 分钟
tensflow2.0系列-反向训练参数问题
TensFlow
在2.0中进行TNN训练反向设置参数时,由于是自定义训练步骤,且每步输入的batch_size = 1 的数据时,在参数设计神经网络中不出现tf.keras.layers.BatchNormalization()时训练效果会更好,如下图所示
tensflow2.0系列-自定义设置单步训练过程
TensFlow
自定义训练训练步骤可以单独设置一个tensflow函数,示例如下: