Merge branch 'main' of https://github.com/sisl/InteractionImitation into main

2021-07-20 05:46:47 -07:00
parent 2e29f42d75 ee7b6b607f
commit c0755f5bb5
3 changed files with 46 additions and 6 deletions
--- a/src/nets/util.py
+++ b/src/nets/util.py
@@ -2,13 +2,11 @@ import torch
 from torch.nn import functional

 def parse_functional(functional_config):
-    if functional_config is None:
-        return None
-    elif isinstance(functional_config, str):
+    if isinstance(functional_config, str):
        if functional_config == 'relu':
            return functional.relu
        elif functional_config == 'sigmoid':
-            return functional.sigmoid
+            return torch.sigmoid
        elif functional_config == 'softmax':
            return functional.softmax
-    
+    return None