VRNN state space motion.
You can do any data augmentation on the training dataset but please leave at least 8 rows for testing purpose.
The model should be able to learning the rolling motion based on training dataset. So once the model sees an object at the location it has never seen (test rows), it should also roll the object from left to right.
Please write comments on what was about my original code.