GPU available: True, used: True TPU available: None, using: 0 TPU cores LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0] initializing ddp: GLOBAL_RANK: 0, MEMBER: 1/1 | Name | Type | Params ----------------------------------------------------------------------------------------- 0 | preprocessor | AudioToMelSpectrogramPreprocessor | 0 1 | preprocessor.featurizer | FilterbankFeatures | 0 2 | encoder | ConvASREncoder | 332 M 3 | encoder.encoder | Sequential | 332 M 4 | encoder.encoder.0 | JasperBlock | 180 K 5 | encoder.encoder.0.mconv | ModuleList | 180 K 6 | encoder.encoder.0.mconv.0 | MaskedConv1d | 180 K 7 | encoder.encoder.0.mconv.0.conv | Conv1d | 180 K 8 | encoder.encoder.0.mconv.1 | BatchNorm1d | 512 9 | encoder.encoder.0.mout | Sequential | 0 10 | encoder.encoder.0.mout.0 | ReLU | 0 11 | encoder.encoder.0.mout.1 | Dropout | 0 12 | encoder.encoder.1 | JasperBlock | 3.7 M 13 | encoder.encoder.1.mconv | ModuleList | 3.6 M 14 | encoder.encoder.1.mconv.0 | MaskedConv1d | 720 K 15 | encoder.encoder.1.mconv.0.conv | Conv1d | 720 K 16 | encoder.encoder.1.mconv.1 | BatchNorm1d | 512 17 | encoder.encoder.1.mconv.3 | Dropout | 0 18 | encoder.encoder.1.mconv.4 | MaskedConv1d | 720 K 19 | encoder.encoder.1.mconv.4.conv | Conv1d | 720 K 20 | encoder.encoder.1.mconv.5 | BatchNorm1d | 512 21 | encoder.encoder.1.mconv.7 | Dropout | 0 22 | encoder.encoder.1.mconv.8 | MaskedConv1d | 720 K 23 | encoder.encoder.1.mconv.8.conv | Conv1d | 720 K 24 | encoder.encoder.1.mconv.9 | BatchNorm1d | 512 25 | encoder.encoder.1.mconv.11 | Dropout | 0 26 | encoder.encoder.1.mconv.12 | MaskedConv1d | 720 K 27 | encoder.encoder.1.mconv.12.conv | Conv1d | 720 K 28 | encoder.encoder.1.mconv.13 | BatchNorm1d | 512 29 | encoder.encoder.1.mconv.15 | Dropout | 0 30 | encoder.encoder.1.mconv.16 | MaskedConv1d | 720 K 31 | encoder.encoder.1.mconv.16.conv | Conv1d | 720 K 32 | encoder.encoder.1.mconv.17 | BatchNorm1d | 512 33 | encoder.encoder.1.res | ModuleList | 66.0 K 34 | encoder.encoder.1.res.0 | ModuleList | 66.0 K 35 | encoder.encoder.1.res.0.0 | MaskedConv1d | 65.5 K 36 | encoder.encoder.1.res.0.0.conv | Conv1d | 65.5 K 37 | encoder.encoder.1.res.0.1 | BatchNorm1d | 512 38 | encoder.encoder.1.mout | Sequential | 0 39 | encoder.encoder.1.mout.1 | Dropout | 0 40 | encoder.encoder.2 | JasperBlock | 3.7 M 41 | encoder.encoder.2.mconv | ModuleList | 3.6 M 42 | encoder.encoder.2.mconv.0 | MaskedConv1d | 720 K 43 | encoder.encoder.2.mconv.0.conv | Conv1d | 720 K 44 | encoder.encoder.2.mconv.1 | BatchNorm1d | 512 45 | encoder.encoder.2.mconv.3 | Dropout | 0 46 | encoder.encoder.2.mconv.4 | MaskedConv1d | 720 K 47 | encoder.encoder.2.mconv.4.conv | Conv1d | 720 K 48 | encoder.encoder.2.mconv.5 | BatchNorm1d | 512 49 | encoder.encoder.2.mconv.7 | Dropout | 0 50 | encoder.encoder.2.mconv.8 | MaskedConv1d | 720 K 51 | encoder.encoder.2.mconv.8.conv | Conv1d | 720 K 52 | encoder.encoder.2.mconv.9 | BatchNorm1d | 512 53 | encoder.encoder.2.mconv.11 | Dropout | 0 54 | encoder.encoder.2.mconv.12 | MaskedConv1d | 720 K 55 | encoder.encoder.2.mconv.12.conv | Conv1d | 720 K 56 | encoder.encoder.2.mconv.13 | BatchNorm1d | 512 57 | encoder.encoder.2.mconv.15 | Dropout | 0 58 | encoder.encoder.2.mconv.16 | MaskedConv1d | 720 K 59 | encoder.encoder.2.mconv.16.conv | Conv1d | 720 K 60 | encoder.encoder.2.mconv.17 | BatchNorm1d | 512 61 | encoder.encoder.2.res | ModuleList | 132 K 62 | encoder.encoder.2.res.0 | ModuleList | 66.0 K 63 | encoder.encoder.2.res.0.0 | MaskedConv1d | 65.5 K 64 | encoder.encoder.2.res.0.0.conv | Conv1d | 65.5 K 65 | encoder.encoder.2.res.0.1 | BatchNorm1d | 512 66 | encoder.encoder.2.res.1 | ModuleList | 66.0 K 67 | encoder.encoder.2.res.1.0 | MaskedConv1d | 65.5 K 68 | encoder.encoder.2.res.1.0.conv | Conv1d | 65.5 K 69 | encoder.encoder.2.res.1.1 | BatchNorm1d | 512 70 | encoder.encoder.2.mout | Sequential | 0 71 | encoder.encoder.2.mout.1 | Dropout | 0 72 | encoder.encoder.3 | JasperBlock | 9.2 M 73 | encoder.encoder.3.mconv | ModuleList | 8.9 M 74 | encoder.encoder.3.mconv.0 | MaskedConv1d | 1.3 M 75 | encoder.encoder.3.mconv.0.conv | Conv1d | 1.3 M 76 | encoder.encoder.3.mconv.1 | BatchNorm1d | 768 77 | encoder.encoder.3.mconv.3 | Dropout | 0 78 | encoder.encoder.3.mconv.4 | MaskedConv1d | 1.9 M 79 | encoder.encoder.3.mconv.4.conv | Conv1d | 1.9 M 80 | encoder.encoder.3.mconv.5 | BatchNorm1d | 768 81 | encoder.encoder.3.mconv.7 | Dropout | 0 82 | encoder.encoder.3.mconv.8 | MaskedConv1d | 1.9 M 83 | encoder.encoder.3.mconv.8.conv | Conv1d | 1.9 M 84 | encoder.encoder.3.mconv.9 | BatchNorm1d | 768 85 | encoder.encoder.3.mconv.11 | Dropout | 0 86 | encoder.encoder.3.mconv.12 | MaskedConv1d | 1.9 M 87 | encoder.encoder.3.mconv.12.conv | Conv1d | 1.9 M 88 | encoder.encoder.3.mconv.13 | BatchNorm1d | 768 89 | encoder.encoder.3.mconv.15 | Dropout | 0 90 | encoder.encoder.3.mconv.16 | MaskedConv1d | 1.9 M 91 | encoder.encoder.3.mconv.16.conv | Conv1d | 1.9 M 92 | encoder.encoder.3.mconv.17 | BatchNorm1d | 768 93 | encoder.encoder.3.res | ModuleList | 297 K 94 | encoder.encoder.3.res.0 | ModuleList | 99.1 K 95 | encoder.encoder.3.res.0.0 | MaskedConv1d | 98.3 K 96 | encoder.encoder.3.res.0.0.conv | Conv1d | 98.3 K 97 | encoder.encoder.3.res.0.1 | BatchNorm1d | 768 98 | encoder.encoder.3.res.1 | ModuleList | 99.1 K 99 | encoder.encoder.3.res.1.0 | MaskedConv1d | 98.3 K 100 | encoder.encoder.3.res.1.0.conv | Conv1d | 98.3 K 101 | encoder.encoder.3.res.1.1 | BatchNorm1d | 768 102 | encoder.encoder.3.res.2 | ModuleList | 99.1 K 103 | encoder.encoder.3.res.2.0 | MaskedConv1d | 98.3 K 104 | encoder.encoder.3.res.2.0.conv | Conv1d | 98.3 K 105 | encoder.encoder.3.res.2.1 | BatchNorm1d | 768 106 | encoder.encoder.3.mout | Sequential | 0 107 | encoder.encoder.3.mout.1 | Dropout | 0 108 | encoder.encoder.4 | JasperBlock | 10.0 M 109 | encoder.encoder.4.mconv | ModuleList | 9.6 M 110 | encoder.encoder.4.mconv.0 | MaskedConv1d | 1.9 M 111 | encoder.encoder.4.mconv.0.conv | Conv1d | 1.9 M 112 | encoder.encoder.4.mconv.1 | BatchNorm1d | 768 113 | encoder.encoder.4.mconv.3 | Dropout | 0 114 | encoder.encoder.4.mconv.4 | MaskedConv1d | 1.9 M 115 | encoder.encoder.4.mconv.4.conv | Conv1d | 1.9 M 116 | encoder.encoder.4.mconv.5 | BatchNorm1d | 768 117 | encoder.encoder.4.mconv.7 | Dropout | 0 118 | encoder.encoder.4.mconv.8 | MaskedConv1d | 1.9 M 119 | encoder.encoder.4.mconv.8.conv | Conv1d | 1.9 M 120 | encoder.encoder.4.mconv.9 | BatchNorm1d | 768 121 | encoder.encoder.4.mconv.11 | Dropout | 0 122 | encoder.encoder.4.mconv.12 | MaskedConv1d | 1.9 M 123 | encoder.encoder.4.mconv.12.conv | Conv1d | 1.9 M 124 | encoder.encoder.4.mconv.13 | BatchNorm1d | 768 125 | encoder.encoder.4.mconv.15 | Dropout | 0 126 | encoder.encoder.4.mconv.16 | MaskedConv1d | 1.9 M 127 | encoder.encoder.4.mconv.16.conv | Conv1d | 1.9 M 128 | encoder.encoder.4.mconv.17 | BatchNorm1d | 768 129 | encoder.encoder.4.res | ModuleList | 445 K 130 | encoder.encoder.4.res.0 | ModuleList | 99.1 K 131 | encoder.encoder.4.res.0.0 | MaskedConv1d | 98.3 K 132 | encoder.encoder.4.res.0.0.conv | Conv1d | 98.3 K 133 | encoder.encoder.4.res.0.1 | BatchNorm1d | 768 134 | encoder.encoder.4.res.1 | ModuleList | 99.1 K 135 | encoder.encoder.4.res.1.0 | MaskedConv1d | 98.3 K 136 | encoder.encoder.4.res.1.0.conv | Conv1d | 98.3 K 137 | encoder.encoder.4.res.1.1 | BatchNorm1d | 768 138 | encoder.encoder.4.res.2 | ModuleList | 99.1 K 139 | encoder.encoder.4.res.2.0 | MaskedConv1d | 98.3 K 140 | encoder.encoder.4.res.2.0.conv | Conv1d | 98.3 K 141 | encoder.encoder.4.res.2.1 | BatchNorm1d | 768 142 | encoder.encoder.4.res.3 | ModuleList | 148 K 143 | encoder.encoder.4.res.3.0 | MaskedConv1d | 147 K 144 | encoder.encoder.4.res.3.0.conv | Conv1d | 147 K 145 | encoder.encoder.4.res.3.1 | BatchNorm1d | 768 146 | encoder.encoder.4.mout | Sequential | 0 147 | encoder.encoder.4.mout.1 | Dropout | 0 148 | encoder.encoder.5 | JasperBlock | 22.0 M 149 | encoder.encoder.5.mconv | ModuleList | 21.2 M 150 | encoder.encoder.5.mconv.0 | MaskedConv1d | 3.3 M 151 | encoder.encoder.5.mconv.0.conv | Conv1d | 3.3 M 152 | encoder.encoder.5.mconv.1 | BatchNorm1d | 1.0 K 153 | encoder.encoder.5.mconv.3 | Dropout | 0 154 | encoder.encoder.5.mconv.4 | MaskedConv1d | 4.5 M 155 | encoder.encoder.5.mconv.4.conv | Conv1d | 4.5 M 156 | encoder.encoder.5.mconv.5 | BatchNorm1d | 1.0 K 157 | encoder.encoder.5.mconv.7 | Dropout | 0 158 | encoder.encoder.5.mconv.8 | MaskedConv1d | 4.5 M 159 | encoder.encoder.5.mconv.8.conv | Conv1d | 4.5 M 160 | encoder.encoder.5.mconv.9 | BatchNorm1d | 1.0 K 161 | encoder.encoder.5.mconv.11 | Dropout | 0 162 | encoder.encoder.5.mconv.12 | MaskedConv1d | 4.5 M 163 | encoder.encoder.5.mconv.12.conv | Conv1d | 4.5 M 164 | encoder.encoder.5.mconv.13 | BatchNorm1d | 1.0 K 165 | encoder.encoder.5.mconv.15 | Dropout | 0 166 | encoder.encoder.5.mconv.16 | MaskedConv1d | 4.5 M 167 | encoder.encoder.5.mconv.16.conv | Conv1d | 4.5 M 168 | encoder.encoder.5.mconv.17 | BatchNorm1d | 1.0 K 169 | encoder.encoder.5.res | ModuleList | 791 K 170 | encoder.encoder.5.res.0 | ModuleList | 132 K 171 | encoder.encoder.5.res.0.0 | MaskedConv1d | 131 K 172 | encoder.encoder.5.res.0.0.conv | Conv1d | 131 K 173 | encoder.encoder.5.res.0.1 | BatchNorm1d | 1.0 K 174 | encoder.encoder.5.res.1 | ModuleList | 132 K 175 | encoder.encoder.5.res.1.0 | MaskedConv1d | 131 K 176 | encoder.encoder.5.res.1.0.conv | Conv1d | 131 K 177 | encoder.encoder.5.res.1.1 | BatchNorm1d | 1.0 K 178 | encoder.encoder.5.res.2 | ModuleList | 132 K 179 | encoder.encoder.5.res.2.0 | MaskedConv1d | 131 K 180 | encoder.encoder.5.res.2.0.conv | Conv1d | 131 K 181 | encoder.encoder.5.res.2.1 | BatchNorm1d | 1.0 K 182 | encoder.encoder.5.res.3 | ModuleList | 197 K 183 | encoder.encoder.5.res.3.0 | MaskedConv1d | 196 K 184 | encoder.encoder.5.res.3.0.conv | Conv1d | 196 K 185 | encoder.encoder.5.res.3.1 | BatchNorm1d | 1.0 K 186 | encoder.encoder.5.res.4 | ModuleList | 197 K 187 | encoder.encoder.5.res.4.0 | MaskedConv1d | 196 K 188 | encoder.encoder.5.res.4.0.conv | Conv1d | 196 K 189 | encoder.encoder.5.res.4.1 | BatchNorm1d | 1.0 K 190 | encoder.encoder.5.mout | Sequential | 0 191 | encoder.encoder.5.mout.1 | Dropout | 0 192 | encoder.encoder.6 | JasperBlock | 23.3 M 193 | encoder.encoder.6.mconv | ModuleList | 22.3 M 194 | encoder.encoder.6.mconv.0 | MaskedConv1d | 4.5 M 195 | encoder.encoder.6.mconv.0.conv | Conv1d | 4.5 M 196 | encoder.encoder.6.mconv.1 | BatchNorm1d | 1.0 K 197 | encoder.encoder.6.mconv.3 | Dropout | 0 198 | encoder.encoder.6.mconv.4 | MaskedConv1d | 4.5 M 199 | encoder.encoder.6.mconv.4.conv | Conv1d | 4.5 M 200 | encoder.encoder.6.mconv.5 | BatchNorm1d | 1.0 K 201 | encoder.encoder.6.mconv.7 | Dropout | 0 202 | encoder.encoder.6.mconv.8 | MaskedConv1d | 4.5 M 203 | encoder.encoder.6.mconv.8.conv | Conv1d | 4.5 M 204 | encoder.encoder.6.mconv.9 | BatchNorm1d | 1.0 K 205 | encoder.encoder.6.mconv.11 | Dropout | 0 206 | encoder.encoder.6.mconv.12 | MaskedConv1d | 4.5 M 207 | encoder.encoder.6.mconv.12.conv | Conv1d | 4.5 M 208 | encoder.encoder.6.mconv.13 | BatchNorm1d | 1.0 K 209 | encoder.encoder.6.mconv.15 | Dropout | 0 210 | encoder.encoder.6.mconv.16 | MaskedConv1d | 4.5 M 211 | encoder.encoder.6.mconv.16.conv | Conv1d | 4.5 M 212 | encoder.encoder.6.mconv.17 | BatchNorm1d | 1.0 K 213 | encoder.encoder.6.res | ModuleList | 1.1 M 214 | encoder.encoder.6.res.0 | ModuleList | 132 K 215 | encoder.encoder.6.res.0.0 | MaskedConv1d | 131 K 216 | encoder.encoder.6.res.0.0.conv | Conv1d | 131 K 217 | encoder.encoder.6.res.0.1 | BatchNorm1d | 1.0 K 218 | encoder.encoder.6.res.1 | ModuleList | 132 K 219 | encoder.encoder.6.res.1.0 | MaskedConv1d | 131 K 220 | encoder.encoder.6.res.1.0.conv | Conv1d | 131 K 221 | encoder.encoder.6.res.1.1 | BatchNorm1d | 1.0 K 222 | encoder.encoder.6.res.2 | ModuleList | 132 K 223 | encoder.encoder.6.res.2.0 | MaskedConv1d | 131 K 224 | encoder.encoder.6.res.2.0.conv | Conv1d | 131 K 225 | encoder.encoder.6.res.2.1 | BatchNorm1d | 1.0 K 226 | encoder.encoder.6.res.3 | ModuleList | 197 K 227 | encoder.encoder.6.res.3.0 | MaskedConv1d | 196 K 228 | encoder.encoder.6.res.3.0.conv | Conv1d | 196 K 229 | encoder.encoder.6.res.3.1 | BatchNorm1d | 1.0 K 230 | encoder.encoder.6.res.4 | ModuleList | 197 K 231 | encoder.encoder.6.res.4.0 | MaskedConv1d | 196 K 232 | encoder.encoder.6.res.4.0.conv | Conv1d | 196 K 233 | encoder.encoder.6.res.4.1 | BatchNorm1d | 1.0 K 234 | encoder.encoder.6.res.5 | ModuleList | 263 K 235 | encoder.encoder.6.res.5.0 | MaskedConv1d | 262 K 236 | encoder.encoder.6.res.5.0.conv | Conv1d | 262 K 237 | encoder.encoder.6.res.5.1 | BatchNorm1d | 1.0 K 238 | encoder.encoder.6.mout | Sequential | 0 239 | encoder.encoder.6.mout.1 | Dropout | 0 240 | encoder.encoder.7 | JasperBlock | 42.9 M 241 | encoder.encoder.7.mconv | ModuleList | 41.3 M 242 | encoder.encoder.7.mconv.0 | MaskedConv1d | 6.9 M 243 | encoder.encoder.7.mconv.0.conv | Conv1d | 6.9 M 244 | encoder.encoder.7.mconv.1 | BatchNorm1d | 1.3 K 245 | encoder.encoder.7.mconv.3 | Dropout | 0 246 | encoder.encoder.7.mconv.4 | MaskedConv1d | 8.6 M 247 | encoder.encoder.7.mconv.4.conv | Conv1d | 8.6 M 248 | encoder.encoder.7.mconv.5 | BatchNorm1d | 1.3 K 249 | encoder.encoder.7.mconv.7 | Dropout | 0 250 | encoder.encoder.7.mconv.8 | MaskedConv1d | 8.6 M 251 | encoder.encoder.7.mconv.8.conv | Conv1d | 8.6 M 252 | encoder.encoder.7.mconv.9 | BatchNorm1d | 1.3 K 253 | encoder.encoder.7.mconv.11 | Dropout | 0 254 | encoder.encoder.7.mconv.12 | MaskedConv1d | 8.6 M 255 | encoder.encoder.7.mconv.12.conv | Conv1d | 8.6 M 256 | encoder.encoder.7.mconv.13 | BatchNorm1d | 1.3 K 257 | encoder.encoder.7.mconv.15 | Dropout | 0 258 | encoder.encoder.7.mconv.16 | MaskedConv1d | 8.6 M 259 | encoder.encoder.7.mconv.16.conv | Conv1d | 8.6 M 260 | encoder.encoder.7.mconv.17 | BatchNorm1d | 1.3 K 261 | encoder.encoder.7.res | ModuleList | 1.6 M 262 | encoder.encoder.7.res.0 | ModuleList | 165 K 263 | encoder.encoder.7.res.0.0 | MaskedConv1d | 163 K 264 | encoder.encoder.7.res.0.0.conv | Conv1d | 163 K 265 | encoder.encoder.7.res.0.1 | BatchNorm1d | 1.3 K 266 | encoder.encoder.7.res.1 | ModuleList | 165 K 267 | encoder.encoder.7.res.1.0 | MaskedConv1d | 163 K 268 | encoder.encoder.7.res.1.0.conv | Conv1d | 163 K 269 | encoder.encoder.7.res.1.1 | BatchNorm1d | 1.3 K 270 | encoder.encoder.7.res.2 | ModuleList | 165 K 271 | encoder.encoder.7.res.2.0 | MaskedConv1d | 163 K 272 | encoder.encoder.7.res.2.0.conv | Conv1d | 163 K 273 | encoder.encoder.7.res.2.1 | BatchNorm1d | 1.3 K 274 | encoder.encoder.7.res.3 | ModuleList | 247 K 275 | encoder.encoder.7.res.3.0 | MaskedConv1d | 245 K 276 | encoder.encoder.7.res.3.0.conv | Conv1d | 245 K 277 | encoder.encoder.7.res.3.1 | BatchNorm1d | 1.3 K 278 | encoder.encoder.7.res.4 | ModuleList | 247 K 279 | encoder.encoder.7.res.4.0 | MaskedConv1d | 245 K 280 | encoder.encoder.7.res.4.0.conv | Conv1d | 245 K 281 | encoder.encoder.7.res.4.1 | BatchNorm1d | 1.3 K 282 | encoder.encoder.7.res.5 | ModuleList | 328 K 283 | encoder.encoder.7.res.5.0 | MaskedConv1d | 327 K 284 | encoder.encoder.7.res.5.0.conv | Conv1d | 327 K 285 | encoder.encoder.7.res.5.1 | BatchNorm1d | 1.3 K 286 | encoder.encoder.7.res.6 | ModuleList | 328 K 287 | encoder.encoder.7.res.6.0 | MaskedConv1d | 327 K 288 | encoder.encoder.7.res.6.0.conv | Conv1d | 327 K 289 | encoder.encoder.7.res.6.1 | BatchNorm1d | 1.3 K 290 | encoder.encoder.7.mout | Sequential | 0 291 | encoder.encoder.7.mout.1 | Dropout | 0 292 | encoder.encoder.8 | JasperBlock | 45.1 M 293 | encoder.encoder.8.mconv | ModuleList | 43.0 M 294 | encoder.encoder.8.mconv.0 | MaskedConv1d | 8.6 M 295 | encoder.encoder.8.mconv.0.conv | Conv1d | 8.6 M 296 | encoder.encoder.8.mconv.1 | BatchNorm1d | 1.3 K 297 | encoder.encoder.8.mconv.3 | Dropout | 0 298 | encoder.encoder.8.mconv.4 | MaskedConv1d | 8.6 M 299 | encoder.encoder.8.mconv.4.conv | Conv1d | 8.6 M 300 | encoder.encoder.8.mconv.5 | BatchNorm1d | 1.3 K 301 | encoder.encoder.8.mconv.7 | Dropout | 0 302 | encoder.encoder.8.mconv.8 | MaskedConv1d | 8.6 M 303 | encoder.encoder.8.mconv.8.conv | Conv1d | 8.6 M 304 | encoder.encoder.8.mconv.9 | BatchNorm1d | 1.3 K 305 | encoder.encoder.8.mconv.11 | Dropout | 0 306 | encoder.encoder.8.mconv.12 | MaskedConv1d | 8.6 M 307 | encoder.encoder.8.mconv.12.conv | Conv1d | 8.6 M 308 | encoder.encoder.8.mconv.13 | BatchNorm1d | 1.3 K 309 | encoder.encoder.8.mconv.15 | Dropout | 0 310 | encoder.encoder.8.mconv.16 | MaskedConv1d | 8.6 M 311 | encoder.encoder.8.mconv.16.conv | Conv1d | 8.6 M 312 | encoder.encoder.8.mconv.17 | BatchNorm1d | 1.3 K 313 | encoder.encoder.8.res | ModuleList | 2.1 M 314 | encoder.encoder.8.res.0 | ModuleList | 165 K 315 | encoder.encoder.8.res.0.0 | MaskedConv1d | 163 K 316 | encoder.encoder.8.res.0.0.conv | Conv1d | 163 K 317 | encoder.encoder.8.res.0.1 | BatchNorm1d | 1.3 K 318 | encoder.encoder.8.res.1 | ModuleList | 165 K 319 | encoder.encoder.8.res.1.0 | MaskedConv1d | 163 K 320 | encoder.encoder.8.res.1.0.conv | Conv1d | 163 K 321 | encoder.encoder.8.res.1.1 | BatchNorm1d | 1.3 K 322 | encoder.encoder.8.res.2 | ModuleList | 165 K 323 | encoder.encoder.8.res.2.0 | MaskedConv1d | 163 K 324 | encoder.encoder.8.res.2.0.conv | Conv1d | 163 K 325 | encoder.encoder.8.res.2.1 | BatchNorm1d | 1.3 K 326 | encoder.encoder.8.res.3 | ModuleList | 247 K 327 | encoder.encoder.8.res.3.0 | MaskedConv1d | 245 K 328 | encoder.encoder.8.res.3.0.conv | Conv1d | 245 K 329 | encoder.encoder.8.res.3.1 | BatchNorm1d | 1.3 K 330 | encoder.encoder.8.res.4 | ModuleList | 247 K 331 | encoder.encoder.8.res.4.0 | MaskedConv1d | 245 K 332 | encoder.encoder.8.res.4.0.conv | Conv1d | 245 K 333 | encoder.encoder.8.res.4.1 | BatchNorm1d | 1.3 K 334 | encoder.encoder.8.res.5 | ModuleList | 328 K 335 | encoder.encoder.8.res.5.0 | MaskedConv1d | 327 K 336 | encoder.encoder.8.res.5.0.conv | Conv1d | 327 K 337 | encoder.encoder.8.res.5.1 | BatchNorm1d | 1.3 K 338 | encoder.encoder.8.res.6 | ModuleList | 328 K 339 | encoder.encoder.8.res.6.0 | MaskedConv1d | 327 K 340 | encoder.encoder.8.res.6.0.conv | Conv1d | 327 K 341 | encoder.encoder.8.res.6.1 | BatchNorm1d | 1.3 K 342 | encoder.encoder.8.res.7 | ModuleList | 410 K 343 | encoder.encoder.8.res.7.0 | MaskedConv1d | 409 K 344 | encoder.encoder.8.res.7.0.conv | Conv1d | 409 K 345 | encoder.encoder.8.res.7.1 | BatchNorm1d | 1.3 K 346 | encoder.encoder.8.mout | Sequential | 0 347 | encoder.encoder.8.mout.1 | Dropout | 0 348 | encoder.encoder.9 | JasperBlock | 74.2 M 349 | encoder.encoder.9.mconv | ModuleList | 71.3 M 350 | encoder.encoder.9.mconv.0 | MaskedConv1d | 12.3 M 351 | encoder.encoder.9.mconv.0.conv | Conv1d | 12.3 M 352 | encoder.encoder.9.mconv.1 | BatchNorm1d | 1.5 K 353 | encoder.encoder.9.mconv.3 | Dropout | 0 354 | encoder.encoder.9.mconv.4 | MaskedConv1d | 14.7 M 355 | encoder.encoder.9.mconv.4.conv | Conv1d | 14.7 M 356 | encoder.encoder.9.mconv.5 | BatchNorm1d | 1.5 K 357 | encoder.encoder.9.mconv.7 | Dropout | 0 358 | encoder.encoder.9.mconv.8 | MaskedConv1d | 14.7 M 359 | encoder.encoder.9.mconv.8.conv | Conv1d | 14.7 M 360 | encoder.encoder.9.mconv.9 | BatchNorm1d | 1.5 K 361 | encoder.encoder.9.mconv.11 | Dropout | 0 362 | encoder.encoder.9.mconv.12 | MaskedConv1d | 14.7 M 363 | encoder.encoder.9.mconv.12.conv | Conv1d | 14.7 M 364 | encoder.encoder.9.mconv.13 | BatchNorm1d | 1.5 K 365 | encoder.encoder.9.mconv.15 | Dropout | 0 366 | encoder.encoder.9.mconv.16 | MaskedConv1d | 14.7 M 367 | encoder.encoder.9.mconv.16.conv | Conv1d | 14.7 M 368 | encoder.encoder.9.mconv.17 | BatchNorm1d | 1.5 K 369 | encoder.encoder.9.res | ModuleList | 3.0 M 370 | encoder.encoder.9.res.0 | ModuleList | 198 K 371 | encoder.encoder.9.res.0.0 | MaskedConv1d | 196 K 372 | encoder.encoder.9.res.0.0.conv | Conv1d | 196 K 373 | encoder.encoder.9.res.0.1 | BatchNorm1d | 1.5 K 374 | encoder.encoder.9.res.1 | ModuleList | 198 K 375 | encoder.encoder.9.res.1.0 | MaskedConv1d | 196 K 376 | encoder.encoder.9.res.1.0.conv | Conv1d | 196 K 377 | encoder.encoder.9.res.1.1 | BatchNorm1d | 1.5 K 378 | encoder.encoder.9.res.2 | ModuleList | 198 K 379 | encoder.encoder.9.res.2.0 | MaskedConv1d | 196 K 380 | encoder.encoder.9.res.2.0.conv | Conv1d | 196 K 381 | encoder.encoder.9.res.2.1 | BatchNorm1d | 1.5 K 382 | encoder.encoder.9.res.3 | ModuleList | 296 K 383 | encoder.encoder.9.res.3.0 | MaskedConv1d | 294 K 384 | encoder.encoder.9.res.3.0.conv | Conv1d | 294 K 385 | encoder.encoder.9.res.3.1 | BatchNorm1d | 1.5 K 386 | encoder.encoder.9.res.4 | ModuleList | 296 K 387 | encoder.encoder.9.res.4.0 | MaskedConv1d | 294 K 388 | encoder.encoder.9.res.4.0.conv | Conv1d | 294 K 389 | encoder.encoder.9.res.4.1 | BatchNorm1d | 1.5 K 390 | encoder.encoder.9.res.5 | ModuleList | 394 K 391 | encoder.encoder.9.res.5.0 | MaskedConv1d | 393 K 392 | encoder.encoder.9.res.5.0.conv | Conv1d | 393 K 393 | encoder.encoder.9.res.5.1 | BatchNorm1d | 1.5 K 394 | encoder.encoder.9.res.6 | ModuleList | 394 K 395 | encoder.encoder.9.res.6.0 | MaskedConv1d | 393 K 396 | encoder.encoder.9.res.6.0.conv | Conv1d | 393 K 397 | encoder.encoder.9.res.6.1 | BatchNorm1d | 1.5 K 398 | encoder.encoder.9.res.7 | ModuleList | 493 K 399 | encoder.encoder.9.res.7.0 | MaskedConv1d | 491 K 400 | encoder.encoder.9.res.7.0.conv | Conv1d | 491 K 401 | encoder.encoder.9.res.7.1 | BatchNorm1d | 1.5 K 402 | encoder.encoder.9.res.8 | ModuleList | 493 K 403 | encoder.encoder.9.res.8.0 | MaskedConv1d | 491 K 404 | encoder.encoder.9.res.8.0.conv | Conv1d | 491 K 405 | encoder.encoder.9.res.8.1 | BatchNorm1d | 1.5 K 406 | encoder.encoder.9.mout | Sequential | 0 407 | encoder.encoder.9.mout.1 | Dropout | 0 408 | encoder.encoder.10 | JasperBlock | 77.3 M 409 | encoder.encoder.10.mconv | ModuleList | 73.7 M 410 | encoder.encoder.10.mconv.0 | MaskedConv1d | 14.7 M 411 | encoder.encoder.10.mconv.0.conv | Conv1d | 14.7 M 412 | encoder.encoder.10.mconv.1 | BatchNorm1d | 1.5 K 413 | encoder.encoder.10.mconv.3 | Dropout | 0 414 | encoder.encoder.10.mconv.4 | MaskedConv1d | 14.7 M 415 | encoder.encoder.10.mconv.4.conv | Conv1d | 14.7 M 416 | encoder.encoder.10.mconv.5 | BatchNorm1d | 1.5 K 417 | encoder.encoder.10.mconv.7 | Dropout | 0 418 | encoder.encoder.10.mconv.8 | MaskedConv1d | 14.7 M 419 | encoder.encoder.10.mconv.8.conv | Conv1d | 14.7 M 420 | encoder.encoder.10.mconv.9 | BatchNorm1d | 1.5 K 421 | encoder.encoder.10.mconv.11 | Dropout | 0 422 | encoder.encoder.10.mconv.12 | MaskedConv1d | 14.7 M 423 | encoder.encoder.10.mconv.12.conv | Conv1d | 14.7 M 424 | encoder.encoder.10.mconv.13 | BatchNorm1d | 1.5 K 425 | encoder.encoder.10.mconv.15 | Dropout | 0 426 | encoder.encoder.10.mconv.16 | MaskedConv1d | 14.7 M 427 | encoder.encoder.10.mconv.16.conv | Conv1d | 14.7 M 428 | encoder.encoder.10.mconv.17 | BatchNorm1d | 1.5 K 429 | encoder.encoder.10.res | ModuleList | 3.6 M 430 | encoder.encoder.10.res.0 | ModuleList | 198 K 431 | encoder.encoder.10.res.0.0 | MaskedConv1d | 196 K 432 | encoder.encoder.10.res.0.0.conv | Conv1d | 196 K 433 | encoder.encoder.10.res.0.1 | BatchNorm1d | 1.5 K 434 | encoder.encoder.10.res.1 | ModuleList | 198 K 435 | encoder.encoder.10.res.1.0 | MaskedConv1d | 196 K 436 | encoder.encoder.10.res.1.0.conv | Conv1d | 196 K 437 | encoder.encoder.10.res.1.1 | BatchNorm1d | 1.5 K 438 | encoder.encoder.10.res.2 | ModuleList | 198 K 439 | encoder.encoder.10.res.2.0 | MaskedConv1d | 196 K 440 | encoder.encoder.10.res.2.0.conv | Conv1d | 196 K 441 | encoder.encoder.10.res.2.1 | BatchNorm1d | 1.5 K 442 | encoder.encoder.10.res.3 | ModuleList | 296 K 443 | encoder.encoder.10.res.3.0 | MaskedConv1d | 294 K 444 | encoder.encoder.10.res.3.0.conv | Conv1d | 294 K 445 | encoder.encoder.10.res.3.1 | BatchNorm1d | 1.5 K 446 | encoder.encoder.10.res.4 | ModuleList | 296 K 447 | encoder.encoder.10.res.4.0 | MaskedConv1d | 294 K 448 | encoder.encoder.10.res.4.0.conv | Conv1d | 294 K 449 | encoder.encoder.10.res.4.1 | BatchNorm1d | 1.5 K 450 | encoder.encoder.10.res.5 | ModuleList | 394 K 451 | encoder.encoder.10.res.5.0 | MaskedConv1d | 393 K 452 | encoder.encoder.10.res.5.0.conv | Conv1d | 393 K 453 | encoder.encoder.10.res.5.1 | BatchNorm1d | 1.5 K 454 | encoder.encoder.10.res.6 | ModuleList | 394 K 455 | encoder.encoder.10.res.6.0 | MaskedConv1d | 393 K 456 | encoder.encoder.10.res.6.0.conv | Conv1d | 393 K 457 | encoder.encoder.10.res.6.1 | BatchNorm1d | 1.5 K 458 | encoder.encoder.10.res.7 | ModuleList | 493 K 459 | encoder.encoder.10.res.7.0 | MaskedConv1d | 491 K 460 | encoder.encoder.10.res.7.0.conv | Conv1d | 491 K 461 | encoder.encoder.10.res.7.1 | BatchNorm1d | 1.5 K 462 | encoder.encoder.10.res.8 | ModuleList | 493 K 463 | encoder.encoder.10.res.8.0 | MaskedConv1d | 491 K 464 | encoder.encoder.10.res.8.0.conv | Conv1d | 491 K 465 | encoder.encoder.10.res.8.1 | BatchNorm1d | 1.5 K 466 | encoder.encoder.10.res.9 | ModuleList | 591 K 467 | encoder.encoder.10.res.9.0 | MaskedConv1d | 589 K 468 | encoder.encoder.10.res.9.0.conv | Conv1d | 589 K 469 | encoder.encoder.10.res.9.1 | BatchNorm1d | 1.5 K 470 | encoder.encoder.10.mout | Sequential | 0 471 | encoder.encoder.10.mout.1 | Dropout | 0 472 | encoder.encoder.11 | JasperBlock | 20.0 M 473 | encoder.encoder.11.mconv | ModuleList | 20.0 M 474 | encoder.encoder.11.mconv.0 | MaskedConv1d | 20.0 M 475 | encoder.encoder.11.mconv.0.conv | Conv1d | 20.0 M 476 | encoder.encoder.11.mconv.1 | BatchNorm1d | 1.8 K 477 | encoder.encoder.11.mout | Sequential | 0 478 | encoder.encoder.11.mout.1 | Dropout | 0 479 | encoder.encoder.12 | JasperBlock | 919 K 480 | encoder.encoder.12.mconv | ModuleList | 919 K 481 | encoder.encoder.12.mconv.0 | MaskedConv1d | 917 K 482 | encoder.encoder.12.mconv.0.conv | Conv1d | 917 K 483 | encoder.encoder.12.mconv.1 | BatchNorm1d | 2.0 K 484 | encoder.encoder.12.mout | Sequential | 0 485 | encoder.encoder.12.mout.1 | Dropout | 0 486 | decoder | ConvASRDecoder | 29.7 K 487 | decoder.decoder_layers | Sequential | 29.7 K 488 | decoder.decoder_layers.0 | Conv1d | 29.7 K 489 | loss | CTCLoss | 0 490 | spec_augmentation | SpectrogramAugmentation | 0 491 | spec_augmentation.spec_cutout | SpecCutout | 0 492 | _wer | WER | 0 ----------------------------------------------------------------------------------------- 332 M Trainable params 0 Non-trainable params 332 M Total params Saving latest checkpoint... Epoch 0, global step 0: val_loss reached 34.24907 (best 34.24907), saving model to "/results/model/checkpoints/finetuned-model---val_loss=34.25-epoch=0.ckpt" as top 3