损失: nan, 使用 WIDER FACES 数据集举例

作者: TonyCongqianWang创建于 2024年6月27日更新于 2024年6月27日

我克隆了您的仓库,使用了 Python 3.6 和 pytorch 1.2。但是,当我使用该模型在您提供的 widerfaces 数据集上进行训练时,仍然得到损失为 nan。我将学习率设置为 1e-3,甚至使用预先训练的模型作为初始化。

内容来源: Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB