forked from huawei/mindspore2022
40 lines
2.8 KiB
ReStructuredText
40 lines
2.8 KiB
ReStructuredText
.. py:class:: mindspore.train.callback.SummaryLandscape(summary_dir)
|
||
|
||
SummaryLandscape可以帮助您收集loss地形图的信息。通过计算loss，可以在PCA（Principal Component Analysis）方向或者随机方向创建地形图。
|
||
|
||
.. note::
|
||
- 使用SummaryLandscape时，需要将代码放置到 `if __name__ == "__main__"` 中运行。
|
||
- SummaryLandscape仅支持Linux系统。
|
||
|
||
**参数：**
|
||
|
||
- **summary_dir** (str) - 该路径将被用来保存创建地形图所使用的数据。
|
||
|
||
|
||
.. py:method:: clean_ckpt()
|
||
|
||
清理checkpoint。
|
||
|
||
.. py:method:: gen_landscapes_with_multi_process(callback_fn, collect_landscape=None, device_ids=None, output=None)
|
||
|
||
使用多进程来生成地形图。
|
||
|
||
**参数：**
|
||
|
||
- **callback_fn** (python function) - Python函数对象，用户需要写一个没有输入的函数，返回值要求如下。
|
||
|
||
- mindspore.train.Model：用户的模型。
|
||
- mindspore.nn.Cell：用户的网络。
|
||
- mindspore.dataset：创建loss所需要的用户数据集。
|
||
- mindspore.nn.Metrics：用户的评估指标。
|
||
|
||
- **collect_landscape** (Union[dict, None]) - 创建loss地形图所用的参数含义与SummaryCollector同名字段一致。此处设置的目的是允许用户可以自由修改创建loss地形图参数。默认值：None。
|
||
|
||
- **landscape_size** (int) - 指定生成loss地形图的图像分辨率。例如：如果设置为128，则loss地形图的分辨率是128*128。计算loss地形图的时间随着分辨率的增大而增加。默认值：40。可选值：3-256。
|
||
- **create_landscape** (dict) - 选择创建哪种类型的loss地形图，分为训练过程loss地形图（train）和训练结果loss地形图（result）。默认值：{"train": True, "result": True}。可选值：True/False。
|
||
- **num_samples** (int) - 创建loss地形图所使用的数据集的大小。例如：在图像数据集中，您可以设置 `num_samples` 是128，这意味着将有128张图片被用来创建loss地形图。注意：`num_samples` 越大，计算loss地形图时间越长。默认值：128。
|
||
- **intervals** (List[List[int]]) - 指定创建loss地形图所需要的checkpoint区间。例如：如果用户想要创建两张训练过程的loss地形图，分别为1-5epoch和6-10epoch，则用户可以设置[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]]。注意：每个区间至少包含3个epoch。
|
||
|
||
- **device_ids** (List(int)) - 指定创建loss地形图所使用的目标设备的ID。例如：[0, 1]表示使用设备0和设备1来创建loss地形图。默认值：None。
|
||
- **output** (str) - 指定保存loss地形图的路径。默认值：None。默认保存路径与summary文件相同。
|