OOM command not allowed when used memory > max memory (solved)

Ray Tune
17

The way to debug here is to comment out different lines in train to see if the shape parameter disappears from ClosureVars.

Note that this is going to help also solve this issue!