8 lines
No EOL
331 B
JavaScript
8 lines
No EOL
331 B
JavaScript
export function disposeUnusedWeightTensors(weightMap, paramMappings) {
|
|
Object.keys(weightMap).forEach(function (path) {
|
|
if (!paramMappings.some(function (pm) { return pm.originalPath === path; })) {
|
|
weightMap[path].dispose();
|
|
}
|
|
});
|
|
}
|
|
//# sourceMappingURL=disposeUnusedWeightTensors.js.map
|