Incorrect lateral deviation

[LHengyi]

I follow the installation instructions starting from a clean ubuntu 18.04 environment.
After running

python run_patch_generation.py data/scenarios/highway/sc1/config_left.json

I check the result in Demo.ipynb. I found the lateral deviation is somehow weird, as the following figure shows

I'm using python 3.7.7. However tensorflow-gpu 1.15 is looking for cuda 10.0 (I tried cuda 10.1 but failed).

I don't see any other error thrown out during the execution. Can anyone help me to figure out the problem?

[LHengyi]

Memory issue.