gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import * as tf from '../../dist/tfjs.esm';
 
import { ConvParams } from '../common/index';
 
export type ScaleLayerParams = {
  weights: tf.Tensor1D
  biases: tf.Tensor1D
}
 
export type ConvLayerParams = {
  conv: ConvParams
  scale: ScaleLayerParams
}
 
export type ResidualLayerParams = {
  conv1: ConvLayerParams
  conv2: ConvLayerParams
}
 
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
}