Fixed bug in options.activation

webmachinelearning · Sep 26, 2023 · 008b0d6 · 008b0d6
1 parent 83737be
commit 008b0d6
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/facial_landmark_detection/ssd_mobilenetv2_face_nchw.js b/facial_landmark_detection/ssd_mobilenetv2_face_nchw.js
@@ -85,6 +85,8 @@ ${nameArray[1]}`;
       } else {
         options.activation = this.builder_.clamp({minValue: 0, maxValue: 6});
       }
+    } else {
+      options.activation = undefined;
     }
     return this.builder_.conv2d(input, weights, options);
   }

diff --git a/facial_landmark_detection/ssd_mobilenetv2_face_nhwc.js b/facial_landmark_detection/ssd_mobilenetv2_face_nhwc.js
@@ -92,6 +92,8 @@ ${nameArray[1]}`;
       } else {
         options.activation = this.builder_.clamp({minValue: 0, maxValue: 6});
       }
+    } else {
+      options.activation = undefined;
     }
     return this.builder_.conv2d(input, weights, options);
   }

diff --git a/image_classification/mobilenet_nhwc.js b/image_classification/mobilenet_nhwc.js
@@ -40,6 +40,8 @@ export class MobileNetV2Nhwc {
       } else {
         options.activation = this.builder_.clamp({minValue: 0, maxValue: 6});
       }
+    } else {
+      options.activation = undefined;
     }
     return this.builder_.conv2d(input, weights, options);
   }

diff --git a/semantic_segmentation/deeplabv3_mnv2_nhwc.js b/semantic_segmentation/deeplabv3_mnv2_nhwc.js
@@ -54,6 +54,8 @@ export class DeepLabV3MNV2Nhwc {
       } else {
         options.activation = this.builder_.clamp({minValue: 0, maxValue: 6});
       }
+    } else {
+      options.activation = undefined;
     }
     return this.builder_.conv2d(input, weights, options);
   }