Fix bug

2020-12-03 20:23:24 +08:00 · 2020-12-03 20:23:24 +08:00 · 1997e2d9db
parent e8752f5c6a
commit 1997e2d9db
1 changed files with 1 additions and 1 deletions
--- a/inference_video.py
+++ b/inference_video.py
@ -157,7 +157,7 @@ for frame in videogen:
    if p > 0.2:
        output = []
        for i in range((2 ** args.exp) - 1):
-            output.append(lastframe.unsqueeze(0))
+            output.append(I0)
    else:
        output = make_inference(I0, I1, args.exp)
    if args.montage: