diff --git a/facial_landmark_detection/ssd_mobilenetv2_face_nchw.js b/facial_landmark_detection/ssd_mobilenetv2_face_nchw.js index 7b7ef186..94ab87ea 100644 --- 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 index 3e9ffd61..e4039dc3 100644 --- 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 index 24391bc8..c5723669 100644 --- 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 index d952051c..2e16c566 100644 --- 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); }