world models and using them to dream drive

You can find the full code here: https://github.com/pandey-ps/world-models/tree/master/drive (it is well commented to explain about the environment of the model). on a higher level, world models are