网络层结构 | 输入输出通道大小 | 输入维度 | 卷积核大小 | 步长 | 输出维度 |
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 |