尝试使用 GeometricFlux.jl 训练可变 GNN

Question

我正在尝试使用 GeometricFlux.jl 训练一个门控图神经网络来预测分子特性：https://github.com/FluxML/GeometricFlux.jl

输入是各种形态的 FeaturedGraph 对象。经过相当多的反复试验，我已经到了异常消息对我不透明的地步，我不知道如何继续。代码：

using Flux, GeometricFlux, MetaGraphs

# convert a MetaGraph to FeaturedGraph
mg = MetaGraph()
adj_mat = adjacency_matrix(mg)
g = FeaturedGraph(adj_mat; nf=ones(2, nv(mg)))

# train the model
model = Chain(GatedGraphConv(2, 1, aggr=:mean), softmax)
loss(x, y) = Flux.mse(model(x), y)
@Flux.epochs 1 Flux.train!(
    loss, 
    Flux.params(model), 
    [(g, 1.0)], 
    ADAM()
)

错误信息是：LoadError: MethodError: _pullback(::Zygote.Context, ::typeof(hcat)) is ambiguous.

任何人都可以提出什么问题吗？