From 95e975878a0d879fb3628ebcd73d7b9b6341e95c Mon Sep 17 00:00:00 2001 From: hzwer <598460606@163.com> Date: Thu, 13 May 2021 17:22:10 +0800 Subject: [PATCH] Update RIFE_HDv3.py --- model/RIFE_HDv3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/RIFE_HDv3.py b/model/RIFE_HDv3.py index 0d7ea3c..3991bd6 100644 --- a/model/RIFE_HDv3.py +++ b/model/RIFE_HDv3.py @@ -199,7 +199,7 @@ class Model: scale_list = [8, 4, 2] flow, _ = self.flownet(imgs, scale_list) res = self.predict(imgs, flow, training=False, UHD=False) - return self.predict(imgs, flow, training=False, UHD=False) + return res def update(self, imgs, gt, learning_rate=0, mul=1, training=True, flow_gt=None): for param_group in self.optimG.param_groups: