added return statement to mlp_

SforAiDl · Sep 3, 2020 · 6f0563e · 6f0563e
1 parent 5067e42
commit 6f0563e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/genrl/utils/utils.py b/genrl/utils/utils.py
@@ -107,6 +107,8 @@ def mlp_(
         layers += [nn.Linear(sizes[layer], sizes[layer + 1]), act]
         weight_init(layers[-1][0].weight)
 
+    return nn.Sequential(*layers)
+
 
 def shared_mlp(
     network1_prev,