out of memory error with dl4j nodes on Knime version 5.4

Before updating to version 5.4 i was able to create a deeplearning workflow with dense layers, now i get a out of memory error, that recommends me to up the off heap memory value, which I’ve done and got the same error.

Any idea why?