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

Apply @tf.function to the __call__ method of RandAugment #310

Closed
LukeWood opened this issue Apr 13, 2022 · 4 comments
Closed

Apply @tf.function to the __call__ method of RandAugment #310

LukeWood opened this issue Apr 13, 2022 · 4 comments

Comments

@LukeWood
Copy link
Contributor

No description provided.

@bhack
Copy link
Contributor

bhack commented Apr 13, 2022

Probably it is better to have some interface to control @tf.function in the base class: https://github.com/keras-team/keras-cv/blob/master/keras_cv/layers/preprocessing/random_augmentation_pipeline.py

So that we could conditionally control also the pipeline compilation similar to keras-team/keras-io#1541

@LukeWood
Copy link
Contributor Author

Removed from v0.1 because MOST users will use this in a tf.data pipeline or keras model that automatically does this.

@LukeWood
Copy link
Contributor Author

I do agree a higher level abstraction would be nice though @bhack

@LukeWood
Copy link
Contributor Author

Closing as tf.data performs this, or the model itself.

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

No branches or pull requests

2 participants