0

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

4

1 回答 1

0

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

于 2020-10-26T18:17:39.803 回答