使用卷积神经网络做训练时遇到了问题

我的任务包含了两个网络,前面有一个MLP(Multilayer Perceptron)网络命名为Model_out,其参数存放在:p_out, re_out = Flux.destructure(model_out),且split_i = length(p_out);第二个网络为CNN,其结构为:model_in = Chain(Ccnv((3,), 2 => N, relu, pad = (0, 1))),其参数存放在:p_in, re_in = Flux.destructure(model_out);两个网络的参数合并为:p = [p_out; p_in];此外我们的任务还需要求解ordinary differentiation equation(ODE);前向传播没有问题,但是当求解ODE后进行反向传播时,出现了报错:“UndefRefError: access to undefined reference;错误定位到NN_in = re_in(p[split_i + 1 : end])(inPut);我在猜想是不是因为CNN的参数排布与MLP不一样,导致我在用索引调用参数矩阵p时,出现了未定义的引用问题,各位大佬能否给点意见,感谢!!

备案号:京ICP备17009874号-2