Memory

[cjxjbd]

When I try to reproduce your code, starting from the second epoch, the memory will be full. Is there any solution? Thanks!

您好，我在尝试复现您的代码的时候，从第二个epoch开始，运存会占满，请问有什么解决办法吗？谢谢！

[cjxjbd]

When using the goosynth dataset.

[gbymat]

When I try to reproduce your code, starting from the second epoch, the memory will be full. Is there any solution? Thanks!

您好，我在尝试复现您的代码的时候，从第二个epoch开始，运存会占满，请问有什么解决办法吗？谢谢！

非常感谢您对我们的工作的关注。
Goosynth数据集所有数据标签被打包成一个单独的pickle文件，里面包含了非常丰富的标注类型，其中场景中每件物品的语义分割标注占了绝大部分存储空间并且在我们的工作中是没有被使用的（我们使用目标检测相关标签）。
您可以尝试将pickle文件进行拆分并修改DataLoader的代码，或者删除标注文件中语义分割部分的相关内容以减少在训练过程中的运行内存的占用问题。
如果显卡显存不够可以尝试缩小batchsize以保证代码可以正常运行。如果有什么其他问题，欢迎提问。