Fixed missing await for builder.build (#238)

webmachinelearning · May 17, 2024 · a014b80 · a014b80
1 parent f4bb7c0
commit a014b80
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/image_classification/resnet50v1_fp16_nchw.js b/image_classification/resnet50v1_fp16_nchw.js
@@ -123,7 +123,7 @@ export class ResNet50V1FP16Nchw {
   }
 
   async build(outputOperand) {
-    this.graph_ = this.builder_.build({'output': outputOperand});
+    this.graph_ = await this.builder_.build({'output': outputOperand});
   }
 
   // Release the constant tensors of a model

diff --git a/image_classification/resnet50v2_nchw.js b/image_classification/resnet50v2_nchw.js
@@ -160,7 +160,7 @@ export class ResNet50V2Nchw {
   }
 
   async build(outputOperand) {
-    this.graph_ = this.builder_.build({'output': outputOperand});
+    this.graph_ = await this.builder_.build({'output': outputOperand});
   }
 
   // Release the constant tensors of a model

diff --git a/image_classification/resnet50v2_nhwc.js b/image_classification/resnet50v2_nhwc.js
@@ -188,7 +188,7 @@ export class ResNet50V2Nhwc {
   }
 
   async build(outputOperand) {
-    this.graph_ = this.builder_.build({'output': outputOperand});
+    this.graph_ = await this.builder_.build({'output': outputOperand});
   }
 
   // Release the constant tensors of a model