我希望批量归一化运行统计数据(均值和方差)在训练结束时收敛,这需要将批量范数动量从一些初始值增加到1.0。我设法使用自定义的Callback改变了动量,但它只有在我的模型在eager模式下编译时才有效。玩具示例(它在epoch 0之后设置momentum=1.0,因为moving_mean应该停止更新): import tensorflow as tf # ve
我有两个ipynb文件:train.ipynb和predict.ipynb。我在train.ipynb中训练了一个带有fit生成器(批处理大小为64)的模型,当我试图在predict.ipynb中加载权重时,捕获到了ResourceExhaustedError。我正在使用tensorflow v1.9和tensorflow docker image中的keras。clear_session() # tried to clear the