Как я могу сохранить мою сессию или мою модель GAN в файл js

Question

Я хочу развернуть свою модель GAN в веб-интерфейсе, для этого мне нужно преобразовать контрольные точки моей модели в js-файлы, которые будут вызываться из веб-кода. Есть функции для save_model и keras для преобразования в pb-файлы, но нет для js

Моя главная проблема в том, что я не понимаю, как записать сессию или переменные веса в js-файлы

Answer 1

Вы можете сохранить модель keras из python.Здесь есть полное учебное руководство здесь , но в основном оно сводится к вызову этого после обучения:

tfjs.converters.save_keras_model(model, tfjs_target_dir)

, а затем размещение результата в общедоступном месте (или на том же сервере, что и веб-интерфейс).Вы можете загрузить свою модель в tenorflow.js следующим образом:

import * as tf from '@tensorflow/tfjs';

const model = await tf.loadLayersModel('https://foo.bar/tfjs_artifacts/model.json');