-
Notifications
You must be signed in to change notification settings - Fork 301
Possible fix for task.summary() #901
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
Conversation
881cfdc
to
4515ea0
Compare
Talk with @fchollet, think we should go for something like this (almost any fix will be a positive delta). Added some tests and marking this ready for review! |
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.
LGTM
) | ||
|
||
|
||
def print_msg(message, line_break=True): |
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.
I wonder if we should export these 2 to internal so our other packages can use them?
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.
I think we would probably still need to copy for now, as we wouldn't want the dep on tf 2.13 anytime soon.
But could be worth thinking about this yeah. Is custom summaries a need for CV or any other keras ecosystem project?
4515ea0
to
07aeb0e
Compare
Make sure model summaries are reportable by the model as soon as they are created, and include a section for preprocessing on the task, if a task has preprocessing.
07aeb0e
to
d6b380e
Compare
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.
Love it, thank you!
I would love to see a model.summary()
call for BertClassifier
in our Getting Started guide soon to show off the work.
Make sure model summaries are reportable by the model as soon as they are created, and include a section for preprocessing on the task, if a task has preprocessing.
Make sure model summaries are reportable by the model as soon as they are created, and include a section for preprocessing on the task, if a task has preprocessing.
Make sure model summaries are reportable by the model as soon as they are created, and include a section for preprocessing on the task, if a task has preprocessing.
See #898.
demo colab