在Tensorflow 2.0中在整个ImageNet上训练类似AlexNet的结构，平坦损失问题

问题描述

我正在尝试在ImageNet上复制AlexNet模型（出于学习目的）。该数据集是120万个ImageNet数据集，并具有5​​0K验证。这是我的代码：

模型功能：

培训功能：

但是，当我开始训练该模型（8个GPU，批处理大小= 32 * nGPU）时，我得到了固定损失和准确性：

你知道造成单位损失的问题是什么吗？

完整代码：here

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）