36 lines
896 B
TypeScript
36 lines
896 B
TypeScript
import * as tf from '@tensorflow/tfjs/dist/tf.es2017.js';
|
|
|
|
import { ConvParams } from '../common';
|
|
|
|
export type ScaleLayerParams = {
|
|
weights: tf.Tensor1D
|
|
biases: tf.Tensor1D
|
|
}
|
|
export type ResidualLayerParams = {
|
|
conv1: ConvLayerParams
|
|
conv2: ConvLayerParams
|
|
}
|
|
|
|
export type ConvLayerParams = {
|
|
conv: ConvParams
|
|
scale: ScaleLayerParams
|
|
}
|
|
|
|
export type NetParams = {
|
|
conv32_down: ConvLayerParams
|
|
conv32_1: ResidualLayerParams
|
|
conv32_2: ResidualLayerParams
|
|
conv32_3: ResidualLayerParams
|
|
conv64_down: ResidualLayerParams
|
|
conv64_1: ResidualLayerParams
|
|
conv64_2: ResidualLayerParams
|
|
conv64_3: ResidualLayerParams
|
|
conv128_down: ResidualLayerParams
|
|
conv128_1: ResidualLayerParams
|
|
conv128_2: ResidualLayerParams
|
|
conv256_down: ResidualLayerParams
|
|
conv256_1: ResidualLayerParams
|
|
conv256_2: ResidualLayerParams
|
|
conv256_down_out: ResidualLayerParams
|
|
fc: tf.Tensor2D
|
|
} |