-
Notifications
You must be signed in to change notification settings - Fork 914
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
Enable TensorFlowModelDataset
to overwrite existing model, and add support for tf.device
#1915
Enable TensorFlowModelDataset
to overwrite existing model, and add support for tf.device
#1915
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.
Took a quick pass; shouldn't have started at 1:30a, so going to leave it at that for now. 😅 Happy to take a look at any responses later.
* Change a file extention to match the previous article Signed-off-by: dinotuku <kuan.tung@epfl.ch> * Add a missing import Signed-off-by: dinotuku <kuan.tung@epfl.ch> * Change both preprocessed datasets to parquet files Signed-off-by: dinotuku <kuan.tung@epfl.ch> * Change data type to ParquetDataSet for parquet files Signed-off-by: dinotuku <kuan.tung@epfl.ch> * Add a note for installing seaborn if it is not installed Signed-off-by: dinotuku <kuan.tung@epfl.ch> Signed-off-by: dinotuku <kuan.tung@epfl.ch> Signed-off-by: William Caicedo <williamc@movio.co>
Signed-off-by: William Caicedo <williamc@movio.co>
Signed-off-by: William Caicedo <williamc@movio.co>
Signed-off-by: William Caicedo <williamc@movio.co>
Signed-off-by: William Caicedo <williamc@movio.co>
Signed-off-by: William Caicedo <williamc@movio.co>
Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: William Caicedo <williamc@movio.co>
321a859
to
8813536
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.
I haven't tested this myself yet, but LGTM from a code review perspective!
Edit: @williamcaicedo I added a note to PR title that added support for tf.device
; can you actually include that in the release notes, too?
TensorFlowModelDataset
to override existing model, and add support for tf.device
TensorFlowModelDataset
to override existing model, and add support for tf.device
TensorFlowModelDataset
to overwrite existing model, and add support for tf.device
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.
Thank you, this PR is very much appreciated⭐️. Haven't tried to run myself, but looking at the test it seems working fine, plus it's not working previously so I am not concern about this.
…icedo/kedro into fix-696-tfmodeldataset
@deepyaman done 👍 |
Amazing, thanks! |
…support for `tf.device` (kedro-org#1915) * Fix issue with save operation. Add gpu option Signed-off-by: William Caicedo <williamc@movio.co> * Add tests Signed-off-by: William Caicedo <williamc@movio.co> * Update RELEASE.md Signed-off-by: William Caicedo <williamc@movio.co> * Update test description Signed-off-by: William Caicedo <williamc@movio.co> * Remove double slash and overwrite flag in fsspec.put method invocation Signed-off-by: William Caicedo <williamc@movio.co> * Allow to explicitly set device name Signed-off-by: William Caicedo <williamc@movio.co> * Update RELEASE.md Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: William Caicedo <williamc@movio.co> * Update docs Signed-off-by: William Caicedo <williamc@movio.co> Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: nickolasrm <nickolasrochamachado@gmail.com>
…support for `tf.device` (kedro-org#1915) * Fix issue with save operation. Add gpu option Signed-off-by: William Caicedo <williamc@movio.co> * Add tests Signed-off-by: William Caicedo <williamc@movio.co> * Update RELEASE.md Signed-off-by: William Caicedo <williamc@movio.co> * Update test description Signed-off-by: William Caicedo <williamc@movio.co> * Remove double slash and overwrite flag in fsspec.put method invocation Signed-off-by: William Caicedo <williamc@movio.co> * Allow to explicitly set device name Signed-off-by: William Caicedo <williamc@movio.co> * Update RELEASE.md Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: William Caicedo <williamc@movio.co> * Update docs Signed-off-by: William Caicedo <williamc@movio.co> Co-authored-by: Deepyaman Datta <deepyaman.datta@utexas.edu> Signed-off-by: Minh Le <m.le@elsevier.com>
Description
prevents TF models from being saved correctly using an unversioned dataset.
Development notes
For models using the SavedModel format, a check was added to find out if the save path already exists. In case it does, path is deleted prior to saving the new model.
Removed the code that manually created all save path intermediate directories if they didn't exist, as
_fs_args.setdefault("auto_mkdir", True)
is already being set.Model uses gpu by default unless
tf_device: cpu
load arg is supplied.Checklist
RELEASE.md
file