OOM happens when running inference.py

[pjwang317]

Hi, after running the train.py, OOM happens when I run inference.py even with several test images, may I ask how to solve this issue? Thank you so much.

[weizhiliang0520]

import os os.environ['CUDA_VISIBLE_DEVICES']='0' setting the gpu ID.