From 1997e2d9db016eea2c3d17909dcbecab9ea4322d Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Thu, 3 Dec 2020 20:23:24 +0800
Subject: [PATCH] Fix bug

---
 inference_video.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference_video.py b/inference_video.py
index ddf2884..f69eb48 100644
--- 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: