search

ctallec / world-models

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

Invalid argument array size #5

Closed Narsil closed 6 years ago

Narsil commented 6 years ago 
RuntimeError: invalid argument 2: size '[-1 x 3 x 96 x 96]' is invalid
for input with 6291456 elements at /pytorch/aten/src/TH/THStorage.c

When trying to run python trainmdrnn.py --logdir exp_dir (After generation and training vae).

Not sure what is going on there.

ctallec commented 6 years ago

I know where this is coming from, I'll fix this asap.

ctallec commented 6 years ago

Should be solved, are you still getting an error?

Narsil commented 6 years ago

It's good now.