Кэширование модели при загрузке - PullRequest
1 голос
/ 28 апреля 2020

В настоящее время я загружаю модель в tfjs со следующим кодом. Я пытаюсь оптимизировать модель, кэшируя файлы ближе к месту, откуда вызывается модель, используя API-интерфейс Cache CloudFlare. Есть ли способ перехватить вызов выборки в операции загрузки для кэширования данных модели и загрузки данных, извлеченных из кэша, в модель?

Текущий код:

const model = await tf.loadLayersModel("model_url_from_gcp")

Попытка достичь:

// Will load data from cache if data is cached, else will fetch from URL and cache
const load_data = cache.get("model_url_from_gcp")

// Will load cached data into Model instead of using a URL 
const model = await tf.loadLayersModel(load_data)

Есть ли способ достичь того же для GraphModels из TFHub?

const model = await tf.loadGraphModel("some_tfhub_url", { fromTFHub: true })

1 Ответ

0 голосов
/ 30 апреля 2020

Для использования LocalStorage или IndexDB для TensorFlow. js см. Эту документацию:

IndexDB: https://www.tensorflow.org/js/guide/save_load#indexeddb_browser_only

LocalStorage: https://www.tensorflow.org/js/guide/save_load#local_storage_browser_only

...