ML.NET图像分类训练冻结

ML.NET Image Classification Training Freezing

我在 Windows 10 PC 上使用 Visual Studio 2019 年的最新版本 ML.NET 图像分类来检测不当图像。我正在使用 3000 个 SFW 和 3000 个 NSFW 图像的数据集来训练它,但它在训练时卡住了。没有错误输出,它只是停止使用 CPU 并停止输出到控制台。

它经常在一行之后随机停止,例如:

[Source=ImageClassificationTrainer; ImageClassificationTrainer, Kind=Trace] Phase: Bottleneck Computation, Dataset used: Train, Image Index: 1109

[Source=ImageClassificationTrainer; MultiClassClassifierScore; Cursor, Kind=Trace] Channel disposed

停止使用 CPU 后,机器学习模型构建器上的训练页面保持不变:

我也尝试过使用每个类型 700 张图像的较小数据集,但最终得到了类似的结果。这是什么原因造成的?

这可能与选择的学习环境有关。您很可能选择了 GPU 训练,但它不受支持。选择CPU.