Update RIFE.py

This commit is contained in:
Zhewei Huang
2022-01-12 14:59:01 +08:00
committed by GitHub
parent 3ed182299c
commit 5b1b69b232

View File

@@ -81,7 +81,6 @@ class Model:
self.optimG.step()
else:
flow_teacher = flow[2]
merged_teacher = merged[2]
return merged[2], {
'merged_tea': merged_teacher,
'mask': mask,