face-api/build/common/extractWeightEntryFactory.js

12 lines
559 B
JavaScript
Raw Normal View History

2020-08-20 02:10:42 +02:00
import { isTensor } from '../utils';
export function extractWeightEntryFactory(weightMap, paramMappings) {
2020-08-18 14:04:33 +02:00
return function (originalPath, paramRank, mappedPath) {
const tensor = weightMap[originalPath];
2020-08-20 02:10:42 +02:00
if (!isTensor(tensor, paramRank)) {
2020-08-18 14:04:33 +02:00
throw new Error(`expected weightMap[${originalPath}] to be a Tensor${paramRank}D, instead have ${tensor}`);
}
paramMappings.push({ originalPath, paramPath: mappedPath || originalPath });
return tensor;
};
}
//# sourceMappingURL=extractWeightEntryFactory.js.map