为此,我想在基于Web的UI上部署GANs模型,我需要将模型的检查点转换为js文件,以供网络代码调用。有用于save_model和keras的函数可以转换为pb文件,而对于js没有此功能
我主要担心的是我很困惑如何在js文件中转储会话或可变权重
答案 0 :(得分:0)
您可以从python保存一个keras模型。有完整的教程here,但基本上等于在训练后称呼它:
driven_by :selenium, using: :selenium_chrome_billy
然后将结果托管在可公开访问的位置(或与Web UI放在同一服务器上),然后可以将模型加载到tensorflow.js中,如下所示:
tfjs.converters.save_keras_model(model, tfjs_target_dir)