extractParamsFromWeigthMap.js 704 B

12345678910111213141516
  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: extractFcParams('fc')
  12. };
  13. disposeUnusedWeightTensors(weightMap, paramMappings);
  14. return { params: params, paramMappings: paramMappings };
  15. }
  16. //# sourceMappingURL=extractParamsFromWeigthMap.js.map