Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix error type

Browse Source
This commit is contained in:
Yin Li 2021-03-11 21:35:09 -05:00
parent c102fc588e
commit 183a223ee6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/models/conv.py
View File

@ -56,7 +56,7 @@ class ConvBlock(nn.Module):
elif l == 'A': elif l == 'A':
return nn.LeakyReLU() return nn.LeakyReLU()
else: else:
raise NotImplementedError('layer type {} not supported'.format(l)) raise ValueError('layer type {} not supported'.format(l))
def _setup_conv(self): def _setup_conv(self):
self.idx_conv += 1 self.idx_conv += 1