add prior_reg flag to FileItemDTO
This commit is contained in:
1 parent
e4ae97e790
commit
e27e229b36
1 file changed
+1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user