gpt4 book ai didi

tensorflow - 在 Firefox 中使用 web worker 训练 LSTM 不适用于 tensorflow js

转载 作者:行者123 更新时间:2023-12-05 07:13:22 25 4
gpt4 key购买 nike

我在训练 LSTM 层时遇到了一个很奇怪的问题。

下面是我的模型。

_________________________________________________________________ 
Layer (type) Output shape Param #
=================================================================
embedding (Embedding) [1,null,256] 131584
_________________________________________________________________
lstm (LSTM) [1,null,1024] 5246976
_________________________________________________________________
lstm_1 (LSTM) [1,null,1024] 8392704
_________________________________________________________________
dense (Dense) [1,null,514] 526850
=================================================================
_________________________________________________________________
  1. 场景 1 - 没有网络 worker ,卡住选项卡。

  2. 场景 2 - 使用网络 worker 进行训练,gfx.offscreencanvas.enabled = false。没有事件,控制台中没有错误日志。没有回调被调用。

  3. 场景 3 - 使用网络 worker 进行训练,gfx.offscreencanvas.enabled = true。该选项卡崩溃,控制台中没有错误日志。没有回调被调用。

注意:

在以下所有场景中,推理都完美无缺。

此外,相同的代码在 chrome 的所有场景中都可以正常工作,没有任何问题。所以我有点困惑为什么它不能在 firefox 上运行。

关于这个问题的任何帮助指导都会非常有帮助。

最佳答案

Firefox 在许多方面与 WebWorker 不兼容...

  • 不支持 Web Worker 中的 WebGL,仅限于 WASM 使用
  • 由于 WebGL 内存处理,使用 WebGL 时它的性能只有 Chrome 的一半(或任何使用 Chromium 的浏览器)

因此,我认为问题在于 WebAssembly (WASM) 由于对计算能力的访问较少而无法编译模型。

关于tensorflow - 在 Firefox 中使用 web worker 训练 LSTM 不适用于 tensorflow js,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60184264/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com