Added relu to nn prescriptor

Author: danyoungday

presp/prescriptor/nn.py

@@ -31,6 +31,8 @@ def __init__(self, model_params: list[dict], device: str = "cpu"):
                 layers.append(torch.nn.Linear(**layer))
             elif layer_type == "tanh":
                 layers.append(torch.nn.Tanh(**layer))
+            elif layer_type == "relu":
+                layers.append(torch.nn.ReLU(**layer))
             elif layer_type == "sigmoid":
                 layers.append(torch.nn.Sigmoid(**layer))
             elif layer_type == "softmax":