From 2ca8e80ee859f7728a8e9190aced2ac26b071b10 Mon Sep 17 00:00:00 2001
From: BowenD-UCB <84425382+BowenD-UCB@users.noreply.github.com>
Date: Tue, 20 Jun 2023 16:06:45 -0700
Subject: [PATCH] changed default non_linearity to silu, matching with
 pretrained model

---
 chgnet/model/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chgnet/model/model.py b/chgnet/model/model.py
index de2d0065..a6b79749 100644
--- a/chgnet/model/model.py
+++ b/chgnet/model/model.py
@@ -51,7 +51,7 @@ def __init__(
         mlp_dropout: float = 0,
         mlp_first: bool = True,
         is_intensive: bool = True,
-        non_linearity: Literal["silu", "relu", "tanh", "gelu"] = "relu",
+        non_linearity: Literal["silu", "relu", "tanh", "gelu"] = "silu",
         atom_graph_cutoff: int = 5,
         bond_graph_cutoff: int = 3,
         cutoff_coeff: int = 5,