Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Multiple Passes through the Same Layer? #356

Open
TonyVT0 opened this issue Nov 3, 2024 · 1 comment
Open

Multiple Passes through the Same Layer? #356

TonyVT0 opened this issue Nov 3, 2024 · 1 comment
Assignees

Comments

@TonyVT0
Copy link

TonyVT0 commented Nov 3, 2024

Hi,

I have a network trained on ai8x layers, but there is one section of the network that requires multiple passes through the same layer. Is it possible to create a YAML file for this type of network that can be converted into C code with the synthesis tool provided?

For a specific example, let us have four intermediate outputs at different feature maps. Is it possible for each of the outputs to be fed into the same layer creating four new outputs? If so, could you provide a sample example for this YAML structure?

Thanks in advanced.

@ermanok ermanok self-assigned this Nov 5, 2024
@ermanok
Copy link
Contributor

ermanok commented Nov 5, 2024

This is possible with weight sharing. This can be done using weight_source parameter in the yaml file. So, if you need to use a layer 4 times in a network, you have to define the same layer 4 times in the yaml file, whose weight_source parameters map to the same layer. You can see the usage of this parameter in FPNDetector example.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants