extractParamsFromWeigthMapTiny.js 672 B

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