mindspore.dataset.debug.DebugHook
- class mindspore.dataset.debug.DebugHook(prev_op_name=None)[源代码]
数据集管道Python调试器钩子的基类。所有用户定义的钩子行为都必须继承这个基类。
为了调试数据集管道中 map 操作的输入和输出数据,用户可以在该类的 compute 函数中添加断点,可以打印日志查看数据的类型和形状等。
- 参数:
prev_op_name (str, 可选) - 上一个调试点的变换名称,默认值:
None
,一般不需指定。
样例:
>>> import mindspore.dataset as ds >>> import mindspore.dataset.debug as debug >>> >>> class CustomizedHook(debug.DebugHook): ... def __init__(self): ... super().__init__() ... ... def compute(self, *args): ... import pdb ... pdb.set_trace() ... print("Data after decode", *args) ... return args >>> >>> # Enable debug mode >>> ds.config.set_debug_mode(True, debug_hook_list=[CustomizedHook()]) >>> >>> # Define dataset pipeline >>> dataset = ds.ImageFolderDataset(dataset_dir="/path/to/image_folder_dataset_directory") >>> # Insert debug hook after `Decode` operation. >>> dataset = dataset.map([vision.Decode(), CustomizedHook(), vision.CenterCrop(100)])