diff --git a/src/super_gradients/conversion/conversion_utils.py b/src/super_gradients/conversion/conversion_utils.py index 2984097e0f..599a9b77bf 100644 --- a/src/super_gradients/conversion/conversion_utils.py +++ b/src/super_gradients/conversion/conversion_utils.py @@ -12,7 +12,7 @@ (torch.int16, np.int16), (torch.int8, np.int8), (torch.uint8, np.uint8), - (torch.bool, np.bool), + (torch.bool, bool), ]