我正在尝试使用 GPU 用我自己的数据训练模型。有 8 类对象需要检测，但是当我开始训练时出现分配错误。
我尝试使用不同的预训练模型，但它继续给我同样的错误。我还尝试减少 .config 文件中的“batch-size”，但它仅在我将 batch-size 等于 1 时才起作用。
我已按照此链接使用自定义数据训练模型：https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#training-the-model
是我的硬件不够用还是别的什么？
GPU：英伟达 GTX 1650
CPU：英特尔 i7 9750H
内存：16 GB
答案 0 :(得分：0)
嗯，你的硬件确实令人印象深刻，我怀疑这会不会有问题。 你可以做的可能是看看其他网站，这是我发现的：
1.https://www.tensorflow.org/tutorials/customization/custom_training_walkthrough 2.https://towardsdatascience.com/custom-object-detection-using-tensorflow-from-scratch-e61da2e10087