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