在培训DITR时，您需要哪些机器规格？（使用变压器端到端对象检测）
我正在研究DITR训练所需的机器规格。 但是，我只有一个GEFORCE 1660 SUPER，并且我遇到了一个“失败”错误。因此，请让我知道您必须使用多少机器规格来完成DITR培训。 请帮助我进行研究。 detr（ https://github.com/facebook beackresearch/detr ）
I am researching the machine specs needed for DETR training.
However, I only have a geforce 1660 super and I got an "out of memory" error. So, please let me know how much machine specs you have to use to complete the DETR training.
Please help me with my research.
DETR(https://github.com/facebookresearch/detr)
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（1）
您已经摆脱了内存错误，因为您的GPU内存不足以支撑批处理大小的输入。尝试以最小批量尺寸运行代码，查看其消耗多少内存，稍微增加批次大小，再次检查内存消耗的增加。这是您将能够估计使用实际批处理大小运行需要多少GPU内存。
我遇到了同样的问题，我切换到具有更大GPU内存（大约24 GB）的机器，然后一切都很好！
Your are getting out of memory error because your GPU memory isn't sufficient to hold up the batch-size you input. Try running the code with minimum batch-size possible, see how much memory it consumes, increase the batch-size slightly, again check increase in memory consumption. This was you will be able to estimate how much GPU memory you require to run it with the actual batch-size.
I had the same issue, I switched to a machine with larger GPU memory (around 24 GB) and then everything worked fine!