Fix error type

2021-03-11 21:35:09 -05:00 · 2021-03-11 21:35:09 -05:00 · 183a223ee6
commit 183a223ee6
parent c102fc588e
1 changed files with 1 additions and 1 deletions
--- 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