123456789101112131415 |
- import { disposeUnusedWeightTensors } from '../common';
- import { loadParamsFactory } from './loadParamsFactory';
- export function extractParamsFromWeigthMap(weightMap) {
- var paramMappings = [];
- var extractDenseBlock4Params = loadParamsFactory(weightMap, paramMappings).extractDenseBlock4Params;
- var params = {
- dense0: extractDenseBlock4Params('dense0', true),
- dense1: extractDenseBlock4Params('dense1'),
- dense2: extractDenseBlock4Params('dense2'),
- dense3: extractDenseBlock4Params('dense3')
- };
- disposeUnusedWeightTensors(weightMap, paramMappings);
- return { params: params, paramMappings: paramMappings };
- }
- //# sourceMappingURL=extractParamsFromWeigthMap.js.map
|