25 lines
745 B
TypeScript
25 lines
745 B
TypeScript
import { extractFCParamsFactory, extractWeightsFactory, ParamMapping } from '../common';
|
|
import { NetParams } from './types';
|
|
|
|
export function extractParams(weights: Float32Array, channelsIn: number, channelsOut: number): { params: NetParams, paramMappings: ParamMapping[] } {
|
|
|
|
const paramMappings: ParamMapping[] = []
|
|
|
|
const {
|
|
extractWeights,
|
|
getRemainingWeights
|
|
} = extractWeightsFactory(weights)
|
|
|
|
const extractFCParams = extractFCParamsFactory(extractWeights, paramMappings)
|
|
|
|
const fc = extractFCParams(channelsIn, channelsOut, 'fc')
|
|
|
|
if (getRemainingWeights().length !== 0) {
|
|
throw new Error(`weights remaing after extract: ${getRemainingWeights().length}`)
|
|
}
|
|
|
|
return {
|
|
paramMappings,
|
|
params: { fc }
|
|
}
|
|
} |