ML.NET 图像分类训练冻结

Question

我在 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 张图像的较小数据集，但最终得到了相似的结果。这是什么原因造成的？

Answer 1

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