ctallec / world-models

Reimplementation of World-Models (Ha and Schmidhuber 2018) in pytorch
MIT License
568 stars 131 forks source link

problem about training VAE #40

Closed Qingcf closed 3 years ago

Qingcf commented 3 years ago

Hi, I am trying to train the VAE (with the step 2 command), and I have generated the datasets by the first command(8 threads and 125 data each thread), but after loading the file buffer, it gives the following error:

File "trainvae.py", line 127, in mkdir(vae_dir) FileNotFoundError: [Errno 2] No such file or directory: 'exp_dir/vae'

Is there anyone facing the same issue? It would be great if you can give me some suggestions about it.

thanks!