|
|
@@ -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) + "'.");
|