steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ensembles

Browse Source
This commit is contained in:
Si11ium 2020-06-14 20:50:53 +02:00
parent ece80ecbed
commit 0cff42f951
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/blocks.py
View File

@ -19,7 +19,8 @@ class LinearModule(ShapeMixin, nn.Module):
def __init__(self, in_shape, out_features, bias=True, activation=None, def __init__(self, in_shape, out_features, bias=True, activation=None,
norm=False, dropout: Union[int, float] = 0, **kwargs): norm=False, dropout: Union[int, float] = 0, **kwargs):
warnings.warn(f'The following arguments have been ignored: \n {list(kwargs.keys())}') if list(kwargs.keys()):
warnings.warn(f'The following arguments have been ignored: \n {list(kwargs.keys())}')
super(LinearModule, self).__init__() super(LinearModule, self).__init__()
self.in_shape = in_shape self.in_shape = in_shape