I am having the same problem, however I have already trained my model with this warning and saved the model.
During inference (testing) I am getting this warning plus OOM error. I wonder if I can solve this for using an already trained model?
Thanks