Added assertion

peiva-git · Dec 16, 2023 · 1c83357 · 1c83357
1 parent 75fb88c
commit 1c83357
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 0 deletions.
diff --git a/basketballtrainer/models/pp_liteseg_rancrops.py b/basketballtrainer/models/pp_liteseg_rancrops.py
@@ -61,6 +61,7 @@ def __init__(self,
             resize_mode=resize_mode,
             pretrained=pretrained
         )
+        assert random_crops >= 0, 'Number of random crops should be a positive integer'
         self.__random_crops = random_crops
         self.__num_classes = num_classes
 

diff --git a/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py b/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py
@@ -63,6 +63,7 @@ def __init__(self,
             resize_mode=resize_mode,
             pretrained=pretrained
         )
+        assert random_crops >= 0, 'Number of random crops should be a positive integer'
         self.__random_crops = random_crops
         self.__num_classes = num_classes
 

diff --git a/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py b/rancrop_model_variants/mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py
@@ -62,6 +62,7 @@ def __init__(self,
             resize_mode=resize_mode,
             pretrained=pretrained
         )
+        assert random_crops >= 0, 'Number of random crops should be a positive integer'
         self.__random_crops = random_crops
 
     def forward(self, x):

diff --git a/...p_model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py b/...p_model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_after_inference.py
@@ -63,6 +63,7 @@ def __init__(self,
             resize_mode=resize_mode,
             pretrained=pretrained
         )
+        assert random_crops >= 0, 'Number of random crops should be a positive integer'
         self.__random_crops = random_crops
         self.__num_classes = num_classes
 

diff --git a/..._model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py b/..._model_variants/softmax_plus_mean_aggregation/pp_liteseg_rancrops_pad_before_inference.py
@@ -63,6 +63,7 @@ def __init__(self,
             resize_mode=resize_mode,
             pretrained=pretrained
         )
+        assert random_crops >= 0, 'Number of random crops should be a positive integer'
         self.__random_crops = random_crops
         self.__num_classes = num_classes