@epfml/discojs-web 3.0.1-p20240902094132.0 → 3.0.1-p20240902162912.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/memory/memory.js +6 -2
- package/package.json +1 -1
package/dist/memory/memory.js
CHANGED
|
@@ -7,8 +7,10 @@
|
|
|
7
7
|
* The working model is loaded from IndexedDB for training (model.fit) only.
|
|
8
8
|
*/
|
|
9
9
|
import { Map } from 'immutable';
|
|
10
|
+
import createDebug from "debug";
|
|
10
11
|
import * as tf from '@tensorflow/tfjs';
|
|
11
12
|
import { Memory, models } from '@epfml/discojs';
|
|
13
|
+
const debug = createDebug('discojs-web:memory');
|
|
12
14
|
export class IndexedDB extends Memory {
|
|
13
15
|
getModelMemoryPath(source) {
|
|
14
16
|
if (typeof source === 'string') {
|
|
@@ -86,7 +88,8 @@ export class IndexedDB extends Memory {
|
|
|
86
88
|
includeOptimizer = false; // true raises an error
|
|
87
89
|
}
|
|
88
90
|
else {
|
|
89
|
-
|
|
91
|
+
debug('unknown working model type %o', model);
|
|
92
|
+
throw new Error(`unknown model type while updating working model`);
|
|
90
93
|
}
|
|
91
94
|
const indexedDBURL = this.getModelMemoryPath(modelInfo);
|
|
92
95
|
await model.extract().save(indexedDBURL, { includeOptimizer });
|
|
@@ -120,7 +123,8 @@ export class IndexedDB extends Memory {
|
|
|
120
123
|
includeOptimizer = false; // true raises an error
|
|
121
124
|
}
|
|
122
125
|
else {
|
|
123
|
-
|
|
126
|
+
debug('unknown saved model type %o', model);
|
|
127
|
+
throw new Error('unknown model type while saving model');
|
|
124
128
|
}
|
|
125
129
|
const indexedDBURL = this.getModelMemoryPath(modelInfo);
|
|
126
130
|
await model.extract().save(indexedDBURL, { includeOptimizer });
|