标签： tensorflow object-detection-api
我正在尝试在自定义数据集上使用Tensorflow的对象检测API训练SSD模型。我已经准备好了tfrecords，并且开始了培训，但是由于我没有足够的RAM而继续被杀。我只有8GB的RAM。我发现更改了batch_size，但这仅更改了我的GPU使用率。有什么方法可以减少RAM使用量？我目前正在使用来自tensorflow / models / research / object_detection / legacy /的train.py。