mindspore.amp.get_black_list
- mindspore.amp.get_black_list()[source]
Provide a copy of internal black list used by auto mixed precision.
The current built-in blacklist contents are:
[
mindspore.nn.BatchNorm1d,mindspore.nn.BatchNorm2d,mindspore.nn.BatchNorm3d,mindspore.nn.LayerNorm]- Returns
list, A copy of internal black list.