From 0cff42f95156170862a0fd8a422ae589935f1295 Mon Sep 17 00:00:00 2001
From: Si11ium <steffen.illium@ifi.lmu.de>
Date: Sun, 14 Jun 2020 20:50:53 +0200
Subject: [PATCH] ensembles
---
modules/blocks.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/modules/blocks.py b/modules/blocks.py
index 9bb0a3b..6b50811 100644
--- 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