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

[Audio] Add AveragePooling1D layer #61

Closed
zaleslaw opened this issue May 18, 2021 · 1 comment · Fixed by #97
Closed

[Audio] Add AveragePooling1D layer #61

zaleslaw opened this issue May 18, 2021 · 1 comment · Fixed by #97
Assignees
Labels
good second issue Good for advanced contributors
Milestone

Comments

@zaleslaw
Copy link
Collaborator

zaleslaw commented May 18, 2021

We are missing some layers to support the export of models from Keras fully. One of them is the AveragePooling1D layer.

Add a layer class, write documentation for it, write a test for it, try, if possible, create a small trainable network with it (in your own GitHub) and attach a link here in the comments.

The layer should be placed here

As a reference implementation, the AvgPool2D layer could be used, but feel free to improve it!

If some refactoring to pooling layers could be applied, please, try to do it!

Also, support for export and import of layer in JSON format should be added (see ModelLoader.kt and ModelSaver.kt)

A detailed description of the layer can be found here

P.S. There are no-ops for tf.nn.avgPool1d in Java API, so you could try to implement it from scratch via available ops or use tf.nn.avgPool like in TensorFlowJS with reshape in tfjs-core/src/ops/avg_pool.ts

@zaleslaw zaleslaw added the good first issue Good for newcomers label May 18, 2021
@zaleslaw zaleslaw added this to the 0.3 milestone May 18, 2021
@zaleslaw zaleslaw added good second issue Good for advanced contributors and removed good first issue Good for newcomers labels May 20, 2021
@mkaze
Copy link
Contributor

mkaze commented Jun 2, 2021

Since this is very similar to MaxPool1D layer, I'll take it and work on it.

@mkaze mkaze mentioned this issue Jun 3, 2021
4 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
good second issue Good for advanced contributors
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants