There seems to be a separate serious leak when using 1.0.0-beta4 or 1.0.0-beta6 version of nd4j.
It does not happen with 1.0.0-b3.
I lowered memory settings to
"-Xms128M",
"-Xmx256M",
"-Dorg.bytedeco.javacpp.maxbytes=256M",
"-Dorg.bytedeco.javacpp.maxphysicalbytes=1G",
and turned off all experience collecting an training, and it still ran out of memory by episode 9.
Probably the problem is with the building of the nodes.
Maybe go back to 1.0.3 and test.
There seems to be a separate serious leak when using 1.0.0-beta4 or 1.0.0-beta6 version of nd4j.
It does not happen with 1.0.0-b3.
I lowered memory settings to
"-Xms128M",
"-Xmx256M",
"-Dorg.bytedeco.javacpp.maxbytes=256M",
"-Dorg.bytedeco.javacpp.maxphysicalbytes=1G",
and turned off all experience collecting an training, and it still ran out of memory by episode 9.
Probably the problem is with the building of the nodes.
Maybe go back to 1.0.3 and test.