-
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 Conv1D impementation #67
Conversation
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.
Great PR, looks like you did a great job, finish some simple review comments and it could be merged
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/Conv2D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/Conv1D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/Conv1D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/Conv2D.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/ConvPadding.kt
Outdated
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/DepthwiseConv2D.kt
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/layer/convolutional/DepthwiseConv2D.kt
Show resolved
Hide resolved
api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/shape/ShapeFunctions.kt
Show resolved
Hide resolved
@avan1235 Please merge the master branch, it helps me to run examples on your branch |
bd4149f
to
e91fa5e
Compare
@zaleslaw I did the rebase and now it should have the newest source code from master |
@avan1235 I recognized that the layer export/import is missing. |
fb69c28
to
865e881
Compare
@zaleslaw thanks for pointing this out, I've now added the proper functions in ModelLoader and ModelSaver |
865e881
to
b9f3567
Compare
LGTM for me, first in the queue to merge at this moment, I'll run the TC build and merge if there are no new issues there. |
Great, just rebased the branch to enable testing |
Resolves #59
I wasn't able to hide the Conv2DImpl class in any way that come to my mind, maybe you have some idea.
My first approach was to create
delegate
field in Conv1D but then there is problem with Layer functions that are not visible and imo should also be then overriden by withdelegate
callsI create a few helper internal functions in ShapeFunctions they maybe useful in more places than the conv tests but I wasn't sure if this is the best place for them.
If implementation and helper function will be ok I can think of some extra tests (I'm open to suggestions)