From e27e229b3664fa5594ae79707593dd6b0d9a31a1 Mon Sep 17 00:00:00 2001 From: squewel Date: Thu, 18 Sep 2025 02:09:39 +0300 Subject: [PATCH] add prior_reg flag to FileItemDTO --- toolkit/data_transfer_object/data_loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/toolkit/data_transfer_object/data_loader.py b/toolkit/data_transfer_object/data_loader.py index b6863663..a7ed3759 100644 --- a/toolkit/data_transfer_object/data_loader.py +++ b/toolkit/data_transfer_object/data_loader.py @@ -121,6 +121,7 @@ class FileItemDTO( self.network_weight: float = self.dataset_config.network_weight self.is_reg = self.dataset_config.is_reg + self.prior_reg = self.dataset_config.prior_reg self.tensor: Union[torch.Tensor, None] = None def cleanup(self):