!5548 Remove unused parameters from SummaryRecord __init__

Merge pull request !5548 from LiHongzhang/rm_unused_params

mindspore/train/summary/_writer_pool.py

@@ -53,7 +53,8 @@ class WriterPool(ctx.Process):
 
     Args:
         base_dir (str): The base directory to hold all the files.
-        filelist (str): The mapping from short name to long filename.
+        max_file_size (Optional[int]): The maximum size of each file that can be written to disk in bytes.
+        filedict (dict): The mapping from plugin to filename.
     """
 
     def __init__(self, base_dir, max_file_size, **filedict) -> None:

mindspore/train/summary/summary_record.py

@@ -75,8 +75,6 @@ class SummaryRecord:
 
     Args:
         log_dir (str): The log_dir is a directory location to save the summary.
-        queue_max_size (int): Deprecated. The capacity of event queue.(reserved). Default: 0.
-        flush_time (int): Deprecated. Frequency of flush the summary file to disk. The unit is second. Default: 120.
         file_prefix (str): The prefix of file. Default: "events".
         file_suffix (str): The suffix of file. Default: "_MS".
         network (Cell): Obtain a pipeline through network for saving graph summary. Default: None.
@@ -84,8 +82,7 @@ class SummaryRecord:
             Unlimited by default. For example, to write not larger than 4GB, specify `max_file_size=4 * 1024**3`.
 
     Raises:
-        TypeError: If the data type of `max_file_size`, `queue_max_size` or `flush_time` is not int, \
-            or the data type of `file_prefix` and `file_suffix` is not str.
+        TypeError: If the type of `max_file_size` is not int, or the type of `file_prefix` or `file_suffix` is not str.
         RuntimeError: If the log_dir is not a normalized absolute path name.
 
     Examples:
@@ -100,14 +97,7 @@ class SummaryRecord:
         >>>     summary_record.close()
     """
 
-    def __init__(self,
-                 log_dir,
-                 queue_max_size=0,
-                 flush_time=120,
-                 file_prefix="events",
-                 file_suffix="_MS",
-                 network=None,
-                 max_file_size=None):
+    def __init__(self, log_dir, file_prefix="events", file_suffix="_MS", network=None, max_file_size=None):
 
         self._closed, self._event_writer = False, None
         self._mode, self._data_pool = 'train', _dictlist()
@@ -120,8 +110,6 @@ class SummaryRecord:
         if not isinstance(max_file_size, (int, type(None))):
             raise TypeError("The 'max_file_size' should be int type.")
 
-        if not isinstance(queue_max_size, int) or not isinstance(flush_time, int):
-            raise TypeError("`queue_max_size` and `flush_time` should be int")
         if not isinstance(file_prefix, str) or not isinstance(file_suffix, str):
             raise TypeError("`file_prefix` and `file_suffix`  should be str.")
 
@@ -129,17 +117,6 @@ class SummaryRecord:
             logger.warning("The 'max_file_size' should be greater than 0.")
             max_file_size = None
 
-        self.queue_max_size = queue_max_size
-        if queue_max_size < 0:
-            # 0 is not limit
-            logger.warning("The queue_max_size(%r) set error, will use the default value: 0", queue_max_size)
-            self.queue_max_size = 0
-
-        self.flush_time = flush_time
-        if flush_time <= 0:
-            logger.warning("The flush_time(%r) set error, will use the default value: 120", flush_time)
-            self.flush_time = 120
-
         self.prefix = file_prefix
         self.suffix = file_suffix
         self.network = network
@@ -155,7 +132,7 @@ class SummaryRecord:
         self._event_writer = WriterPool(log_dir,
                                         max_file_size,
                                         summary=self.full_file_name,
-                                        lineage=get_event_file_name('events', '_lineage'))
+                                        lineage=get_event_file_name(self.prefix, '_lineage'))
         _get_summary_tensor_data()
         atexit.register(self.close)