Deploying to gh-pages from @ b81e927 🚀

webmachinelearning · Jun 11, 2024 · 59981a4 · 59981a4
1 parent 9f8d7f9
commit 59981a4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/face_recognition/facenet_nchw.js b/face_recognition/facenet_nchw.js
@@ -41,9 +41,7 @@ export class FaceNetNchw {
         bias: bias,
       };
     }
-    if (relu) {
-      options.activation = this.builder_.relu();
-    }
+
     // WebNN spec drops autoPad support, compute the explicit padding instead.
     if (options.autoPad == 'same-upper') {
       options.padding =
@@ -52,7 +50,8 @@ export class FaceNetNchw {
             /* oihw */[weights.shape()[2], weights.shape()[3]],
             options.strides, options.dilations, options.autoPad);
     }
-    return this.builder_.conv2d(input, weights, options);
+    const conv2d = this.builder_.conv2d(input, weights, options);
+    return relu ? this.builder_.relu(conv2d) : conv2d;
   }
 
   async buildBlock35_(input, indice, weightsSuffixes) {