2020-08-20 02:10:42 +02:00
|
|
|
import { extractFCParamsFactory, extractWeightsFactory } from '../common';
|
|
|
|
export function extractParams(weights, channelsIn, channelsOut) {
|
2020-08-18 14:04:33 +02:00
|
|
|
const paramMappings = [];
|
2020-08-20 02:10:42 +02:00
|
|
|
const { extractWeights, getRemainingWeights } = extractWeightsFactory(weights);
|
|
|
|
const extractFCParams = extractFCParamsFactory(extractWeights, paramMappings);
|
2020-08-18 14:04:33 +02:00
|
|
|
const fc = extractFCParams(channelsIn, channelsOut, 'fc');
|
|
|
|
if (getRemainingWeights().length !== 0) {
|
|
|
|
throw new Error(`weights remaing after extract: ${getRemainingWeights().length}`);
|
|
|
|
}
|
|
|
|
return {
|
|
|
|
paramMappings,
|
|
|
|
params: { fc }
|
|
|
|
};
|
|
|
|
}
|
|
|
|
//# sourceMappingURL=extractParams.js.map
|