From d6e55ebec6e8e7d6746931b269308f9f145212e9 Mon Sep 17 00:00:00 2001
From: im_yeong_jae <iyj0121@ajou.ac.kr>
Date: Wed, 3 May 2023 12:43:24 +0900
Subject: [PATCH] loss change

---
 src/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/trainer.py b/src/trainer.py
index 07d31a1..9c2e153 100644
--- a/src/trainer.py
+++ b/src/trainer.py
@@ -52,7 +52,7 @@ class Trainer():
             res, sr = self.model(lr, 0)
             t_res, _ = self.t_model(lr, 0)
             kd_loss = self.KD_loss(res, t_res)
-            loss = self.loss(sr, hr) + 0.1*kd_loss
+            loss = self.loss(sr, hr) + 0.01*kd_loss
             loss.backward()
             if self.args.gclip > 0:
                 utils.clip_grad_value_(
-- 
GitLab