github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add prior_reg flag to FileItemDTO

Browse Source
This commit is contained in:
squewel
2025-09-18 02:09:39 +03:00
committed by GitHub
parent e4ae97e790
commit e27e229b36
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
toolkit/data_transfer_object/data_loader.py
View File

@@ -121,6 +121,7 @@ class FileItemDTO(
self.network_weight: float = self.dataset_config.network_weight self.network_weight: float = self.dataset_config.network_weight
self.is_reg = self.dataset_config.is_reg self.is_reg = self.dataset_config.is_reg
self.prior_reg = self.dataset_config.prior_reg
self.tensor: Union[torch.Tensor, None] = None self.tensor: Union[torch.Tensor, None] = None
def cleanup(self): def cleanup(self):