Merge pull request #76 from rishishian/master

fix a little bug in weight_drop.py
PetrochukM · Jul 12, 2019 · 09c0acd · 09c0acd
2 parents f57af13 + eb28005
commit 09c0acd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchnlp/nn/weight_drop.py b/torchnlp/nn/weight_drop.py
@@ -21,7 +21,7 @@ def forward(*args, **kwargs):
             w = torch.nn.functional.dropout(raw_w, p=dropout, training=module.training)
             setattr(module, name_w, w)
 
-        return original_module_forward(*args)
+        return original_module_forward(*args, **kwargs)
 
     setattr(module, 'forward', forward)