extractParamsFromWeigthMap.js 716 B

123456789101112131415
  1. import { disposeUnusedWeightTensors } from '../common';
  2. import { loadParamsFactory } from './loadParamsFactory';
  3. export function extractParamsFromWeigthMap(weightMap) {
  4. var paramMappings = [];
  5. var extractDenseBlock4Params = loadParamsFactory(weightMap, paramMappings).extractDenseBlock4Params;
  6. var params = {
  7. dense0: extractDenseBlock4Params('dense0', true),
  8. dense1: extractDenseBlock4Params('dense1'),
  9. dense2: extractDenseBlock4Params('dense2'),
  10. dense3: extractDenseBlock4Params('dense3')
  11. };
  12. disposeUnusedWeightTensors(weightMap, paramMappings);
  13. return { params: params, paramMappings: paramMappings };
  14. }
  15. //# sourceMappingURL=extractParamsFromWeigthMap.js.map