mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-16 16:37:51 +01:00
Update RIFE.py
This commit is contained in:
@@ -81,7 +81,6 @@ class Model:
|
|||||||
self.optimG.step()
|
self.optimG.step()
|
||||||
else:
|
else:
|
||||||
flow_teacher = flow[2]
|
flow_teacher = flow[2]
|
||||||
merged_teacher = merged[2]
|
|
||||||
return merged[2], {
|
return merged[2], {
|
||||||
'merged_tea': merged_teacher,
|
'merged_tea': merged_teacher,
|
||||||
'mask': mask,
|
'mask': mask,
|
||||||
|
|||||||
Reference in New Issue
Block a user