[Error] Can not train VAE about world-models HOT 6 CLOSED

Have you generated the dataset using the first command ? If not, first generate the dataset, then launch the second command. Otherwise, can you check that your datasets/carracing folder is not empty (beware, there's an s at the end of dataset, if you generated in dataset/carracing, this might be the cause of your issue) ?

from world-models.

I generated the dataset with the same command structure, but used 500 rollouts and 64 threads. And yes, I meant datasets with "s". So the path to my *.npz is "world-models/datasets/carracing/thread_x/rollout_y.npz", where x goes from 0 to 63, and y goes from 0 to 7.

Should I run the first command with 1000 rollouts?

Thanks!

from world-models.

Ok, this is probably the problem, you need to generate at least 601 rollouts ... That's a bit hardcoded, but basically we are using the first 600 rollout files to train and the rest to test. If you generate at least 800 rollouts (to have a decent test set) you should be fine.

from world-models.

Alright, I will try to get >601 and report back!

from world-models.

Small correction, we use the 600 lasts as test set, and the rest as train set.

from world-models.

The VAE is finally training, thanks!

from world-models.