diff --git a/test.py b/test.py new file mode 100644 index 0000000..236bb91 --- /dev/null +++ b/test.py @@ -0,0 +1,8 @@ +# python .\test.py +from __init__ import VideoReverseNode +import torch +frames = torch.arange(12).view(4, 3) # pretend we have 4 frames +print("Before:", frames) +node = VideoReverseNode() +out = node.reverse(frames) +print("After:", out[0])