extractParamsFromWeigthMap.js 894 B

12345678910111213141516171819202122
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var common_1 = require("../common");
  4. function extractParamsFromWeigthMap(weightMap) {
  5. var paramMappings = [];
  6. var extractWeightEntry = common_1.extractWeightEntryFactory(weightMap, paramMappings);
  7. function extractFcParams(prefix) {
  8. var weights = extractWeightEntry(prefix + "/weights", 2);
  9. var bias = extractWeightEntry(prefix + "/bias", 1);
  10. return { weights: weights, bias: bias };
  11. }
  12. var params = {
  13. fc: {
  14. age: extractFcParams('fc/age'),
  15. gender: extractFcParams('fc/gender')
  16. }
  17. };
  18. common_1.disposeUnusedWeightTensors(weightMap, paramMappings);
  19. return { params: params, paramMappings: paramMappings };
  20. }
  21. exports.extractParamsFromWeigthMap = extractParamsFromWeigthMap;
  22. //# sourceMappingURL=extractParamsFromWeigthMap.js.map