Custom Flux Layer Connect

Specific Domains Machine Learning
,
1

Hello everyone,

I would like to implement the architecture of N-BEATS network (https://arxiv.org/pdf/1905.10437.pdf) in julia Flux. A simple example of a part of the architecture is shown in the following figure.

NBeats_part
NBeats_part846×743 30.9 KB

Is there a way to define the connections between the layers individually? For example connect one layer with 64 neurons to two other layers (96 and 32 neurons). The Flux.SkipConnection method does not seem to work here, as I only want to connect the input values to the first output of Block #1.

2

Hi, did you manage to complete the implementation?

(looking to do the same thing)

3

I would just create a custom type containing all layers and define its forward pass. See the docs Custom Layers · Flux

4

I did not find a solution and stopped back then. @CarloLucibello the solution sounds promising