mindspore.amp.get_black_list

mindspore.amp.get_black_list()[源代码]

提供用于自动混合精度的内置黑名单的拷贝。

当前的内置黑名单内容为:

[mindspore.nn.BatchNorm1d, mindspore.nn.BatchNorm2d, mindspore.nn.BatchNorm3d, mindspore.nn.LayerNorm]

返回:

list:内置黑名单的拷贝。