Fix error type

This commit is contained in:
Yin Li 2021-03-11 21:35:09 -05:00
parent c102fc588e
commit 183a223ee6

View File

@ -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