From cfc373651fdd2e055140b0f597b0745bc1b2e96d Mon Sep 17 00:00:00 2001 From: Kenta Sato Date: Mon, 11 Mar 2024 23:00:28 +0000 Subject: [PATCH] add functor --- src/MessagePassingIPA.jl | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/MessagePassingIPA.jl b/src/MessagePassingIPA.jl index fd2383b..459e9a7 100644 --- a/src/MessagePassingIPA.jl +++ b/src/MessagePassingIPA.jl @@ -310,6 +310,8 @@ struct GeometricVectorPerceptronGNN gvpstack::Chain end +Flux.@functor GeometricVectorPerceptronGNN + """ GeometricVectorPerceptronGNN( (sn, vn),