From abbc086dd768e4ec1ee03682e97ede73a133d3e2 Mon Sep 17 00:00:00 2001 From: sammlapp Date: Thu, 12 Sep 2024 18:00:19 -0400 Subject: [PATCH] catch any exception when checking channel dimensions --- opensoundscape/ml/cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opensoundscape/ml/cnn.py b/opensoundscape/ml/cnn.py index c9917160..1efc110a 100644 --- a/opensoundscape/ml/cnn.py +++ b/opensoundscape/ml/cnn.py @@ -633,7 +633,7 @@ def __init__( f"(originally {self.preprocessor.channels})." ) self.preprocessor.channels = arch_channels - except ChannelDimCheckError: + except: # can we try to check if first layer expects input with channels=channels? warnings.warn( f"Failed to detect expected # input channels of this architecture."