extractParamsFromWeigthMap.js 787 B

12345678910111213141516171819
  1. import { disposeUnusedWeightTensors, extractWeightEntryFactory } from '../common';
  2. export function extractParamsFromWeigthMap(weightMap) {
  3. var paramMappings = [];
  4. var extractWeightEntry = extractWeightEntryFactory(weightMap, paramMappings);
  5. function extractFcParams(prefix) {
  6. var weights = extractWeightEntry(prefix + "/weights", 2);
  7. var bias = extractWeightEntry(prefix + "/bias", 1);
  8. return { weights: weights, bias: bias };
  9. }
  10. var params = {
  11. fc: {
  12. age: extractFcParams('fc/age'),
  13. gender: extractFcParams('fc/gender')
  14. }
  15. };
  16. disposeUnusedWeightTensors(weightMap, paramMappings);
  17. return { params: params, paramMappings: paramMappings };
  18. }
  19. //# sourceMappingURL=extractParamsFromWeigthMap.js.map