loadParamsFactory.d.ts 387 B

123456
  1. import { ParamMapping } from '../common';
  2. import { DenseBlock3Params, DenseBlock4Params } from './types';
  3. export declare function loadParamsFactory(weightMap: any, paramMappings: ParamMapping[]): {
  4. extractDenseBlock3Params: (prefix: string, isFirstLayer?: boolean) => DenseBlock3Params;
  5. extractDenseBlock4Params: (prefix: string, isFirstLayer?: boolean) => DenseBlock4Params;
  6. };