Memory issue

Hi,

After running pre-trained model for albu solution (batch size 1 and only one image(#230) is placed as data set), the docker memory rapidly grows and ultimately system is hanged. After some time, the process is auto killed by OS.
Please advise what is the resolution of this issue. I have my Ubuntu machine 18.04 with 64G physical memory.

Thanks,
Ajmal
