Merge pull request #88 from MendelXu/main

add assert terms

ssod/datasets/samplers/semi_sampler.py

@@ -21,6 +21,8 @@ def __init__(
         rank=None,
         **kwargs
     ):
+        # check to avoid some problem
+        assert samples_per_gpu > 1, "samples_per_gpu should be greater than 1."
         _rank, _num_replicas = get_dist_info()
         if num_replicas is None:
             num_replicas = _num_replicas
@@ -190,6 +192,7 @@ def __len__(self):
 
     def set_epoch(self, epoch):
         self.epoch = epoch
+
     # duplicated, implement it by weight instead of sampling
     # def update_sample_ratio(self):
     #     if self.dynamic_step is not None: