search

XiYe20 / VPTR

The repository for paper VPTR: Efficient Transformers for Video Prediction
MIT License
88 stars 19 forks source link

Dataset np array key issue #4

Closed May-226 closed 1 year ago

May-226 commented 1 year ago

When running the notebook, the MovingMNISTDataset dataloader will be called, and the way to create dictionary 'data' is based on the key of the input .npz file, however, the original dataset downloaded from http://www.cs.toronto.edu/~nitish/unsupervised_video/ seems not to have the key 'clips'. Is it possible for you to share your splitted dataset somehow? Thx!

XiYe20 commented 1 year ago

Hi, I use the movingmnist dataset provided by this paper: https://github.com/google/e3d_lstm. Sorry for the inconvenience.

May-226 commented 1 year ago

Thanks for the prompt reply! It works now.