tweak

lucidrains · lucidrains · commit cffb43108966 · 2022-11-28T12:13:30.000-08:00
diff --git a/equiformer_pytorch/equiformer_pytorch.py b/equiformer_pytorch/equiformer_pytorch.py
@@ -707,8 +707,7 @@ def __init__(
         self.to_rel_dist_hidden = nn.Sequential(
             nn.Linear(1, radial_hidden_dim),
             nn.SiLU(),
-            LayerNorm(radial_hidden_dim),
-            nn.Linear(radial_hidden_dim, radial_hidden_dim),
+            LayerNorm(radial_hidden_dim)
         )
 
         # whether to differentiate through basis, needed gradients for iterative refinement
diff --git a/equiformer_pytorch/version.py b/equiformer_pytorch/version.py
@@ -1,3 +1,3 @@
-__version__ = '0.0.24'
+__version__ = '0.0.25'
 
 __cuda_pkg_name__ = f'equiformer_pytorch_cuda_{__version__.replace(".", "_")}'

Original file line number	Diff line number	Diff line change
`@@ -707,8 +707,7 @@ def __init__(`
`707`	`707`	`self.to_rel_dist_hidden = nn.Sequential(`
`708`	`708`	`nn.Linear(1, radial_hidden_dim),`
`709`	`709`	`nn.SiLU(),`
`710`		`- LayerNorm(radial_hidden_dim),`
`711`		`- nn.Linear(radial_hidden_dim, radial_hidden_dim),`
	`710`	`+ LayerNorm(radial_hidden_dim)`
`712`	`711`	`)`
`713`	`712`
`714`	`713`	`# whether to differentiate through basis, needed gradients for iterative refinement`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`		`-__version__ = '0.0.24'`
	`1`	`+__version__ = '0.0.25'`
`2`	`2`
`3`	`3`	`__cuda_pkg_name__ = f'equiformer_pytorch_cuda_{__version__.replace(".", "_")}'`