我的硬件是否足以训练对象检测模型？
Is my hardware enough to train an object detection model?
我正在尝试使用 GPU 使用我自己的数据训练模型。有 8 类 个对象要检测，但是当我开始训练时出现分配错误。
我试过使用不同的预训练模型，但它仍然给我同样的错误。我还尝试减少 .config 文件中的“批量大小”，但只有当我将批量大小等于 1 时它才有效。
我已经按照这个 link 训练了一个带有自定义数据的模型：https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#training-the-model
是我的硬件不够还是别的什么？
显卡：英伟达 GTX 1650
CPU：英特尔 i7 9750H
内存：16GB
嗯，您的硬件确实令人印象深刻，我怀疑这会有问题。
你可以做的可能是看看其他网站，这是我发现的：
1.https://www.tensorflow.org/tutorials/customization/custom_training_walkthrough
2.https://towardsdatascience.com/custom-object-detection-using-tensorflow-from-scratch-e61da2e10087
我正在尝试使用 GPU 使用我自己的数据训练模型。有 8 类 个对象要检测，但是当我开始训练时出现分配错误。
我试过使用不同的预训练模型，但它仍然给我同样的错误。我还尝试减少 .config 文件中的“批量大小”，但只有当我将批量大小等于 1 时它才有效。
我已经按照这个 link 训练了一个带有自定义数据的模型：https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#training-the-model
是我的硬件不够还是别的什么？
显卡：英伟达 GTX 1650
CPU：英特尔 i7 9750H
内存：16GB
嗯，您的硬件确实令人印象深刻，我怀疑这会有问题。 你可以做的可能是看看其他网站，这是我发现的：
1.https://www.tensorflow.org/tutorials/customization/custom_training_walkthrough 2.https://towardsdatascience.com/custom-object-detection-using-tensorflow-from-scratch-e61da2e10087