support ctrl shuffle for seg

paddlex/repo_apis/PaddleSeg_api/base_seg_config.py

@@ -73,6 +73,11 @@ class BaseSegConfig(BaseConfig):
                           bool), "print_mem_info should be a bool"
         self.set_val('print_mem_info', print_mem_info)
 
+    def update_shuffle(self, shuffle: bool):
+        """setting print memory info"""
+        assert isinstance(shuffle, bool), "shuffle should be a bool"
+        self.set_val('shuffle', shuffle)
+
     def update_pretrained_weights(self, weight_path, is_backbone=False):
         """ update_pretrained_weights """
         if 'model' not in self:

paddlex/repo_apis/PaddleSeg_api/seg/model.py

@@ -132,6 +132,7 @@ class SegModel(BaseModel):
             config.update_log_ranks(device)
             amp = benchmark.get('amp', None)
             config.update_print_mem_info(benchmark.get('print_mem_info', True))
+            config.update_shuffle(benchmark.get('shuffle', False))
             if repeats is not None:
                 assert isinstance(repeats, int), 'repeats must be an integer.'
                 cli_args.append(CLIArgument('--repeats', repeats))