Comments (5)
Hi @PowerToThePeople111 ,
I don't think the problem in saving is actually coming from model subclassing. Most probably, it comes from passing df
as argument to the call
method.
By slightly changing your code to pass a list instead of a Pandas dataframe, the model can be saved.
class DataPrep2(tf.keras.Model):
def __init__(self):
super().__init__()
self.layer1 = tf.keras.layers.StringLookup(output_mode='int', vocabulary=["a", "b"])
self.layer2 = tf.keras.layers.Discretization(bin_boundaries=[0,1,2,3,4], output_mode='int')
def call(self, x):
return {
"conv1": self.layer1(x[0]),
"conv2": self.layer2(x[1])
}
dp2 = DataPrep2()
dp2([df["var1"], df["var2"]])
dp2.save("delteme")
from tensorflow.
@PowerToThePeople111 Please refer to the TensorFlow documentation for the latest recommendations on saving subclassed models: https://www.tensorflow.org/tutorials/keras/save_and_load and kindly provide the access to your notebook.
Thank you!
from tensorflow.
Hi sushreebarsa,
thank you for your reply. You should now be able to access the notebook. Also I put the whole code neccessary below the link to it.
I read the documentation again and was lead by the following parts:
You can switch to the SavedModel format by:
Passing save_format='tf' to save()
Passing a filename without an extension...
*Custom objects (for example, subclassed models or layers) require special attention when saving and loading. Refer to the Saving custom objects section below.
...
So for me it seemed that the above is basically everything i need to do since later on in the "Custom objects"-section you can read at the very top:
If you are using the SavedModel format, you can skip this section.
So actually I should be able to store the model by using save()
with save_format 'tf'
and providing a filename without ending, which is what I did:
Neither dp2.save("delteme", save_format="tf")
nor dp2.save("delteme")
did the trick for me and result in the error message i posted in my initial statement.
Did I miss something here?
from tensorflow.
Thank you @karimsr4,
that solution works. :)
from tensorflow.
Are you satisfied with the resolution of your issue?
Yes
No
from tensorflow.
Related Issues (20)
- Model.add_loss is not documented HOT 1
- Cannot use @TaskAction annotation on method IncrementalTask.taskAction$gradle_core() because interface org.gradle.api.tasks.incremental.IncrementalTaskInputs is not a valid parameter to an action method. HOT 4
- ERROR: @local_config_cuda//:enable_cuda :: Error loading option @local_config_cuda//:enable_cuda: 'NoneType' value has no field or method 'replace' HOT 2
- A checker is needed for inputs of Conv layers. HOT 1
- [Bazel CI] Bazel build error: Target '//tensorflow/tools/pip_package:build_pip_package' contains an error with Bazel@HEAD HOT 1
- TensorFlow lite compilation error under Ubuntu 22.04.4 LTS HOT 2
- Upgrading TFLite package (from TF 2.8 to Play Services) HOT 1
- Unknown: JIT Compilation Failed
- ValueError: Invalid dtype: str768 HOT 1
- Infinity loop when try to save model with input_signatures on funtion decorator HOT 3
- Tensorflow does not recognice GPU no matter what HOT 2
- __dict__ error when saving a model using tf.saved_model.save(model, modelSavedPath)
- softplus on zero values is incorrect HOT 3
- We integrate tensorflow movenet with our webapp, and it stopped working (it worked two weeks ago) with the following error message, what can be the possible reasons and how to fix it? HOT 2
- Tensorflow TF ops - CPU Factory not registered HOT 8
- Complex dtype input for keras layer in tf2.16+ HOT 1
- [Android x86_64] [TFLite] `Fatal signal 11 (SIGSEGV), code 128 (SI_KERNEL), fault addr 0x0` crash from `libtensorflowlite.so` HOT 1
- [DOCS] Missing complex input for Round op HOT 1
- Ensemble provides either correct or (the same) incorrect inference results HOT 2
- tf.keras.utils.plot_model doesn't work HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from tensorflow.