Lutz Roeder 3 лет назад
Родитель
Сommit
69a3a70d88
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      source/tflite.js

+ 1 - 1
source/tflite.js

@@ -382,7 +382,7 @@ tflite.Node = class {
                 for (const entry of Object.entries(options)) {
                     const name = entry[0];
                     const value = entry[1];
-                    if (name === 'fused_activation_function' && value !== 0) {
+                    if (name === 'fused_activation_function' && value) {
                         const activationFunctionMap = { 1: 'Relu', 2: 'ReluN1To1', 3: 'Relu6', 4: 'Tanh', 5: 'SignBit' };
                         if (!activationFunctionMap[value]) {
                             throw new tflite.Error("Unsupported activation funtion index '" + JSON.stringify(value) + "'.");