From 1415f009572fe91952f6cbd29dfea06efbfa8aad Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Wed, 27 Apr 2022 11:47:55 +0800
Subject: [PATCH] Update RIFE.py

---
 model/RIFE.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/RIFE.py b/model/RIFE.py
index c7c09e8..d3d7e6f 100644
--- a/model/RIFE.py
+++ b/model/RIFE.py
@@ -76,7 +76,7 @@ class Model:
         loss_tea = (self.lap(merged_teacher, gt)).mean()
         if training:
             self.optimG.zero_grad()
-            loss_G = loss_l1 + loss_tea + loss_distill * 0.01
+            loss_G = loss_l1 + loss_tea + loss_distill * 0.01 # when training RIFEm, the weight of loss_distill should be 0.005 or 0.002
             loss_G.backward()
             self.optimG.step()
         else: