add functor

MurrellGroup · Mar 11, 2024 · cfc3736 · cfc3736
1 parent ed26b45
commit cfc3736
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/MessagePassingIPA.jl b/src/MessagePassingIPA.jl
@@ -310,6 +310,8 @@ struct GeometricVectorPerceptronGNN
     gvpstack::Chain
 end
 
+Flux.@functor GeometricVectorPerceptronGNN
+
 """
     GeometricVectorPerceptronGNN(
         (sn, vn),