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。