danijar / dreamer

Dream to Control: Learning Behaviors by Latent Imagination
https://danijar.com/dreamer
MIT License
506 stars 109 forks source link

AttributeError: 'MirroredStrategy' object has no attribute 'experimental_run_v2' #50

Closed lylyjy closed 3 years ago

lylyjy commented 3 years ago

@tf.function() def train(self, data, log_images=False):

strategy.run

self._strategy.experimental_run_v2(self._train, args=(data, log_images))
danijar commented 3 years ago

You're using a newer version of TF than is listed in the readme. The method is called run() now.

That said, I highly recommend giving DreamerV2 a try. The code base also supports Gaussian latents with a flag.