當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Tensorflow.js tf.enableProdMode()用法及代碼示例


Tensorflow.js是由Google開發的開源庫,用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.enableProdMode() 函數用於啟用生產模式,該生產模式停用支持生產的精確性限製。

用法:

tf.enableProdMode()

參數:該方法不持有任何參數。

返回值:它返回void。



範例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling enableProdMode() method
await tf.enableProdMode();
  
// Setting debug mode of the
// environment
tf.env().set('DEBUG', false);
  
// Printing output
console.log(tf.env().flags);

輸出:

{
  "IS_BROWSER":true,
  "IS_NODE":false,
  "DEBUG":false,
  "CPU_HANDOFF_SIZE_THRESHOLD":128,
  "PROD":true,
  "WEBGL_VERSION":2,
  "HAS_WEBGL":true,
  "WEBGL_CHECK_NUMERICAL_PROBLEMS":false,
  "IS_TEST":false,
  "WEBGL_CPU_FORWARD":true,
  "WEBGL_MAX_TEXTURE_SIZE":16384,
  "WEBGL_FORCE_F16_TEXTURES":true,
  "WEBGL_RENDER_FLOAT32_CAPABLE":true,
  "WEBGL_RENDER_FLOAT32_ENABLED":true,
  "WEBGL_FLUSH_THRESHOLD":-1,
  "WEBGL_PACK":true,
  "WEBGL_LAZILY_UNPACK":true,
  "WEBGL_DELETE_TEXTURE_THRESHOLD":-1,
  "WEBGL_PACK_BINARY_OPERATIONS":true,
  "WEBGL_USE_SHAPES_UNIFORMS":false,
  "WEBGL_PACK_UNARY_OPERATIONS":true,
  "WEBGL_DOWNLOAD_FLOAT_ENABLED":true,
  "WEBGL_CONV_IM2COL":true,
  "WEBGL_PACK_DEPTHWISECONV":true,
  "WEBGL_MAX_TEXTURES_IN_SHADER":16,
  "WEBGL_PACK_ARRAY_OPERATIONS":true
}

範例2:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling enableProdMode() method
await tf.enableProdMode();
  
// Setting prod mode of the environment
tf.env().set("PROD", !0)
  
// Setting textures of the environment
tf.env().set('WEBGL_FORCE_F16_TEXTURES', false);
  
// Calling ready() method
await tf.ready();
  
// Printing output
console.log(tf.env().features);

輸出:

{
  "IS_BROWSER":true,
  "IS_NODE":false,
  "DEBUG":false,
  "CPU_HANDOFF_SIZE_THRESHOLD":128,
  "PROD":true,
  "WEBGL_VERSION":2,
  "HAS_WEBGL":true,
  "WEBGL_CHECK_NUMERICAL_PROBLEMS":false,
  "IS_TEST":false,
  "WEBGL_CPU_FORWARD":true,
  "WEBGL_MAX_TEXTURE_SIZE":16384,
  "WEBGL_FORCE_F16_TEXTURES":false,
  "WEBGL_RENDER_FLOAT32_CAPABLE":true,
  "WEBGL_RENDER_FLOAT32_ENABLED":true,
  "WEBGL_FLUSH_THRESHOLD":-1,
  "WEBGL_PACK":true,
  "WEBGL_LAZILY_UNPACK":true,
  "WEBGL_DELETE_TEXTURE_THRESHOLD":-1,
  "WEBGL_PACK_BINARY_OPERATIONS":true,
  "WEBGL_USE_SHAPES_UNIFORMS":false,
  "WEBGL_PACK_UNARY_OPERATIONS":true,
  "WEBGL_DOWNLOAD_FLOAT_ENABLED":true,
  "WEBGL_CONV_IM2COL":true,
  "WEBGL_PACK_DEPTHWISECONV":true,
  "WEBGL_MAX_TEXTURES_IN_SHADER":16,
  "WEBGL_PACK_ARRAY_OPERATIONS":true
}

參考: https://js.tensorflow.org/api/latest/#enableProdMode




相關用法


注:本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.enableProdMode() Function。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。