enforced that loadParametersBuffer takes in a buffer that matches the…

… number of parameters
microsoft · Oct 25, 2023 · cd2e4fa · cd2e4fa
1 parent c74112e
commit cd2e4fa
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/js/common/lib/training-session-impl.ts b/js/common/lib/training-session-impl.ts
@@ -155,6 +155,11 @@ export class TrainingSession implements TrainingSessionInterface {
   }
 
   async loadParametersBuffer(array: Float32Array, trainableOnly: boolean): Promise<void> {
+    const paramsSize = await this.getParametersSize(trainableOnly);
+    if (array.length !== paramsSize) {
+      throw new Error('Size of the buffer passed into loadParametersBuffer must match the number of parameters in ' +
+        'the model. Please use getParametersSize method to check.');
+    }
     return this.handler.loadParametersBuffer(array, trainableOnly);
   }