ensembles

2020-06-14 20:50:53 +02:00
parent ece80ecbed
commit 0cff42f951
1 changed files with 2 additions and 1 deletions
--- a/modules/blocks.py
+++ b/modules/blocks.py
@@ -19,7 +19,8 @@ class LinearModule(ShapeMixin, nn.Module):

    def __init__(self, in_shape, out_features, bias=True, activation=None,
                 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__()

        self.in_shape = in_shape