Skip to content
This repository has been archived by the owner on Oct 17, 2021. It is now read-only.

Add sanity check on batchSize in Model evaluate, fit & predict #302

Merged
merged 3 commits into from
Aug 27, 2018

Conversation

caisq
Copy link
Contributor

@caisq caisq commented Aug 25, 2018

Also fix a bug in Model.fit(), in which if the method call
errors out early on, the Model.isTraining bit is not flipped
back to false.

Fixes: tensorflow/tfjs#641


This change is Reviewable

Also fix a bug in Model.fit(), in which if the method call
errors out early on, the Model.isTraining bit is not flipped
back to false.
Copy link
Contributor

@dsmilkov dsmilkov left a comment

Choose a reason for hiding this comment

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

Thank you!

Reviewed 2 of 2 files at r1.
Reviewable status: 0 of 1 approvals obtained (waiting on @nsthorat, @dsmilkov, and @bileschi)

@caisq
Copy link
Contributor Author

caisq commented Aug 27, 2018

Thanks for the review!

@caisq caisq merged commit a15211c into tensorflow:master Aug 27, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
2 participants