diff --git a/mindspore/ccsrc/dataset/core/config_manager.cc b/mindspore/ccsrc/dataset/core/config_manager.cc index a489b4a4ce8f18e3b91296d126921b1811a05b77..9291a8f832de2c5d69ead553cb3e490aa9b5ae96 100644 --- a/mindspore/ccsrc/dataset/core/config_manager.cc +++ b/mindspore/ccsrc/dataset/core/config_manager.cc @@ -41,6 +41,7 @@ Status ConfigManager::FromJson(const nlohmann::json &j) { set_worker_connector_size(j.value("workerConnectorSize", worker_connector_size_)); set_op_connector_size(j.value("opConnectorSize", op_connector_size_)); set_seed(j.value("seed", seed_)); + set_monitor_sampling_interval(j.value("monitorSamplingInterval", monitor_sampling_interval_)); return Status::OK(); } diff --git a/mindspore/dataset/core/configuration.py b/mindspore/dataset/core/configuration.py index d3175cd1819196b250e3d02c2a4d174a2cbcbd71..5376c668c40dbe5c1eb35d55e57b62f5c888ebac 100644 --- a/mindspore/dataset/core/configuration.py +++ b/mindspore/dataset/core/configuration.py @@ -185,7 +185,8 @@ class ConfigurationManager: >>> # "numParallelWorkers": 4, >>> # "workerConnectorSize": 16, >>> # "opConnectorSize": 16, - >>> # "seed": 5489 + >>> # "seed": 5489, + >>> # "monitorSamplingInterval": 30 >>> # } """ self.config.load(file) diff --git a/tests/ut/data/dataset/declient.cfg b/tests/ut/data/dataset/declient.cfg index b657ead6d5fe46d4416132e8dfa3cf0eb3faaabb..36ca5ec3d96d3df7c46e47bb33604bdb48e1134f 100644 --- a/tests/ut/data/dataset/declient.cfg +++ b/tests/ut/data/dataset/declient.cfg @@ -4,6 +4,6 @@ "numParallelWorkers": 4, "workerConnectorSize": 16, "opConnectorSize": 16, - "seed": 5489 - + "seed": 5489, + "monitorSamplingInterval": 15 } diff --git a/tests/ut/python/dataset/test_config.py b/tests/ut/python/dataset/test_config.py index c4d665b3917f9e7bf98c535b3df4de013f9b3cd9..09da76356395659868f692951401facbb8182399 100644 --- a/tests/ut/python/dataset/test_config.py +++ b/tests/ut/python/dataset/test_config.py @@ -45,6 +45,7 @@ def test_basic(): # assert ds.config.get_worker_connector_size() == 16 assert ds.config.get_prefetch_size() == 16 assert ds.config.get_seed() == 5489 + assert ds.config.get_monitor_sampling_interval() == 15 # ds.config.set_rows_per_buffer(1) ds.config.set_num_parallel_workers(2)