fix #issue 881, nb4reg_over_task_ratio is 0 cause error

marrlab · Oct 10, 2024 · 5eefec0 · 5eefec0
1 parent fe84b5b
commit 5eefec0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/domainlab/algos/trainers/a_trainer.py b/domainlab/algos/trainers/a_trainer.py
@@ -194,13 +194,17 @@ def before_tr(self):
         self.cal_reg_loss_over_task_loss_ratio()
 
     def cal_reg_loss_over_task_loss_ratio(self):
+        """
+        estimate the scale of each loss term, match each loss term to the major
+        loss via a ratio, this ratio will be multiplied with multiplier
+        """
         list_accum_reg_loss = []
         loss_task_agg = 0
         for ind_batch, (tensor_x, tensor_y, tensor_d, *others) in enumerate(
             self.loader_tr
         ):
             if ind_batch >= self.aconf.nb4reg_over_task_ratio:
-                break
+                return
             tensor_x, tensor_y, tensor_d = (
                 tensor_x.to(self.device),
                 tensor_y.to(self.device),

diff --git a/sh_list_error.sh b/sh_list_error.sh