-
Notifications
You must be signed in to change notification settings - Fork 103
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
Add GlobalAvgPool1D Layer #74
Conversation
The unit tests written right now are only for GlobalAvgPool1D extending the PoolingTest class. With the help of helper functions written by @avan1235 in the ShapeFunctions |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So, this LGTM for me also I'd like the tests, it's a great implementation and very clear for now
I suggest removing some blank lines in some places, but it's ready for merge.
There is a queue of PRs, I'll notify you when you should merge master into your branch (in a few days) and your PR will be merged
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/GlobalAvgPool1D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/GlobalAvgPool1D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/pooling/GlobalAvgPool1D.kt
Outdated
Show resolved
Hide resolved
api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/PoolLayerTest.kt
Outdated
Show resolved
Hide resolved
api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/PoolLayerTest.kt
Outdated
Show resolved
Hide resolved
api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/GlobalAvgPooling1DTest.kt
Outdated
Show resolved
Hide resolved
api/src/test/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/PoolLayerTest.kt
Outdated
Show resolved
Hide resolved
So you could merge the master branch and I'll merge your PR in two days |
aad5f01
to
73b48df
Compare
Congrats, @therealansh thanks for the contribution! |
This PR adds GlobalAvgPool1D layer (#63). Also renamed the file to "GlobalAvgPooling2D" from "GlovalAvgPooling2d"(typo).
TODO: