Changed the network architecture a bit and tuned the hyper-params. Run is more stable.
mentioned in commit d647ba0e
merged