Load and save model issue about tfjs HOT 5 OPEN

Hi, @Infiniteskys

Thank you for bringing this issue to our attention and please give it try with below code snippet and see is it working as expected or not ? Please refer official documentation of asyncStorageIO to save and load the model to/from async storage

import * as tf from "@tensorflow/tfjs";
import { asyncStorageIO } from "@tensorflow/tfjs-react-native";

export const loadModel = async () => {
  try {
    await tf.ready();
    const model = await tf.loadGraphModel("https://test_model/model.json");
    console.log("Model loaded.");
    model.summary();

    // Save the model to AsyncStorage
    const savedModel = await model.save(asyncStorageIO('customs'));
    console.log("Model saved to AsyncStorage");

    // Load the model from AsyncStorage
    const loadedModel = await tf.loadLayersModel(asyncStorageIO('customs'));
    console.log(loadedModel == model); // This might not be true due to model serialization

    return loadedModel;

  } catch (e) {
    console.error("Error:", e);
    return null;
  }
};

If issue still persists please let us know and if possible please help us with your Github repo along with model and complete steps to run code to replicate the same behaviour from our end.

Thank you for your cooperation and patience.

from tfjs.

Hi @gaikwadrahul8 ,
I had tried it but it still not working and show the error that layer improper config format.
I believe the json file we convert is a graph model, I had tried to use Loadgraphmodel function, but I can only load from the Website URL but not the asynstrongIO.

from tfjs.

If I have the function as below, it will show the error: Based on the provided shape, [4.2],the tensor should have 8 values but has 1.

import * as tf from "@tensorflow/tfjs";
import { asyncStorageIO } from "@tensorflow/tfjs-react-native";

export const loadModel = async () => {
try {
await tf.ready();
const model = await tf.loadGraphModel("https://test_model/model.json");
console.log("Model loaded.");

// Save the model to AsyncStorage
const savedModel = await model.save(asyncStorageIO('customs'));
console.log("Model saved to AsyncStorage");

// Load the model from AsyncStorage
const loadedModel = await tf.loadGraphModel(asyncStorageIO('customs'));

return loadedModel;

} catch (e) {
console.error("Error:", e);
return null;
}
};

from tfjs.

Hi, @Infiniteskys

If possible could you please help us with your model file in zip format along with complete steps to replicate the same behavior from our end to investigate this issue further ?

Thank you for your cooperation and patience.

from tfjs.

Hi @gaikwadrahul8 ,
Here is the test model other friend train at the service side and convert it at the service.
test_model.zip

I had also test to use the location directly.

const loadedModel = await tf.loadLayersModel('/moblientclient/assest/model/test_model/model.json');

But it will show error the network request fail. But in this case, I dont have any idea how to save the model if there is a update version at the service.

Thank you.

from tfjs.