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

Fix initializers import and export #348

Merged
merged 10 commits into from
May 18, 2022

Conversation

juliabeliaeva
Copy link
Contributor

@juliabeliaeva juliabeliaeva commented Apr 7, 2022

This PR fixes intializers import and export, plus adds some of the missing initializers: Constant, Ones, Zeros, Orthogonal and TruncatedNormal. This will fix #331.

@juliabeliaeva juliabeliaeva requested a review from zaleslaw April 7, 2022 15:59
@juliabeliaeva juliabeliaeva force-pushed the initializer-fixes branch 2 times, most recently from 4149c70 to 5116e4c Compare April 29, 2022 17:33
@juliabeliaeva juliabeliaeva marked this pull request as ready for review May 9, 2022 19:12
import org.jetbrains.kotlinx.dl.api.core.layer.core.Input
import org.junit.jupiter.api.Test

class InitializerImportExportTest {
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

One optional enhancement that comes to mind is to add tests comparing outputs of original models and deserialized ones.

@juliabeliaeva juliabeliaeva merged commit f9e1eac into Kotlin:master May 18, 2022
@juliabeliaeva juliabeliaeva deleted the initializer-fixes branch May 18, 2022 20:25
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

Successfully merging this pull request may close these issues.

Add export in Keras style for a few new initializers
2 participants