网络层结构

输入输出通道大小

输入维度

卷积核大小

步长

输出维度

Conv2d_1

7 × 16

B × 7 × T × 257

3 × 3

(1, 2)

B × 16 × T × 128

Conv2d_2

16 × 16

B × 16 × T × 128

3 × 3

(1, 2)

B × 16 × T × 63

Conv2d_3

16 × 32

B × 16 × T × 63

3 × 3

(1, 2)

B × 32 × T × 31

Conv2d_4

32 × 32

B × 32 × T × 31

3 × 3

(1, 2)

B × 32 × T × 15

Conv2d_5

32 × 64

B × 32 × T × 15

3 × 3

(1, 2)

B × 64 × T × 7

Reshape

——

B × 64 × T × 7

——

——

B × T × 448

LSTM_1

——

B × T × 448

——

——

B × T × 448

LSTM_2

——

B × T × 448

——

——

B × T × 448

Reshape

——

B × T × 448

——

——

B × 64 × T × 7

TransConv2d_1

128 × 32

B × 128 × T × 7

3 × 3

(1, 2)

B × 32 × T × 15

TransConv2d_2

64 × 32

B × 64 × T × 15

3 × 3

(1, 2)

B × 32 × T × 31

TransConv2d_3

64 × 16

B × 64 × T × 31

3 × 3

(1, 2)

B × 16 × T × 63

TransConv2d_4

32 × 16

B × 32 × T × 63

3 × 3

(1, 2)

B × 16 × T × 128

TransConv2d_5

32 × 6

B × 32 × T × 257

3 × 3

(1, 2)

B × 6 × T × 257