From 183a223ee64d0c6668f4c29bcc7ff188096ef0e9 Mon Sep 17 00:00:00 2001 From: Yin Li Date: Thu, 11 Mar 2021 21:35:09 -0500 Subject: [PATCH] Fix error type --- map2map/models/conv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/map2map/models/conv.py b/map2map/models/conv.py index b4d9146..523d0ff 100644 --- a/map2map/models/conv.py +++ b/map2map/models/conv.py @@ -56,7 +56,7 @@ class ConvBlock(nn.Module): elif l == 'A': return nn.LeakyReLU() else: - raise NotImplementedError('layer type {} not supported'.format(l)) + raise ValueError('layer type {} not supported'.format(l)) def _setup_conv(self): self.idx_conv += 1