search

sangmin-git / LMC-Memory

Official PyTorch implementation of "Video Prediction Recalling Long-term Motion Context via Memory Alignment Learning" (CVPR 2021 Oral)
Other
86 stars 22 forks source link

Input 10 frames to predict 20 frames #17

Open LeiShe1 opened 2 years ago

LeiShe1 commented 2 years ago

Question: I typed 10 frames to predict 20 frames, so short_len was set to 10, long_len to 20, and out_len to 20, but I ran into problems and couldn't train

LeiShe1 commented 2 years ago

image

LeiShe1 commented 2 years ago

I set batch_size to 48, and the program works, but set batch_size > 48, the above error will appear.

huyi1998 commented 2 years ago

I set batch_size to 48, and the program works, but set batch_size > 48, the above error will appear.

Do you solve it?