从检查点加载循环时出错

File ~/srrf/srrf.py:104, in make_trainer(model, train_data, test_data, loss, metrics, name, optimizer, schedule, steps_per_cp, eval_batch_size) 94 train_task = trax.supervised.training.TrainTask(labeled_data=train_data, 95 labeled_data=train_data, 96 labeled_data=train_data, 97 labeled_data=train_data, 98 labeled_data=train_data, 99 labeled_data=train_data, 100 labeled_data=train_data, 101 labeled_data=train_data, 102 labeled_data=train_data, 103 labeled_data=train_data, 104 labeled_data=train_data, 105 labeled_data=train_data, 106 labeled_data=train_data, 107 labeled_data=train_data, 108 labeled_data=train_data, 109 labeled_data=train_data, 110 labeled_data=train_data, 111 labeled_data=train_data, 112 labeled_data=train_data, 113 labeled_data=train_data, 114 labeled_data=train_data, 115 labeled_data=train_data, 116 labeled_data=train_data, 117 labeled_data=train_data, 118 labeled_data=train_data, 119 labeled_data=train_data, 120 labeled_data=train_data, 121 labeled_data=train_data, 122 labeled_data=train_data, 123 labeled_data=train_data, 124 labeled_data=train_data, 125 labeled_data=train_data, 126 labeled_data=train_data, 127 labeled_data=train_data, 128 labeled_data=train_data, 129 labeled_data=train_data, 130 labeled_data=train_data, 131 labeled_data=train_data, 132 labeled_data=train_data, 133 labeled_data=train_data, 134 labeled_data=train_data, 135 labeled_data=train_data, 136 labeled_data=train_data, 137 labeled_data=train_data, 138 labeled_data=train_data, 139 labeled_data=train_data, 140 labeled_data=train_data, 141 labeled_data=train_data, 142 labeled_data=train_data, 143 labeled_data=train_data, 144 labeled_data=train_data, 145 labeled_data=train_data, 146 labeled_data=train_data, 147 labeled_data=train_data, 148 labeled_data=train_data, 149 labeled_data=train_data, 150 labeled_data=train_data, 151 labeled_data=train_data, 152 labeled_data=train_data, 153 labeled_data=train_data, 154 labeled_data=train_data, 155 labeled_data=train_data, 156 labeled_data=train_data, 157 labeled_data=train_data, 158 labeled_data=train_data, 159 labeled_data=train_data, 160 labeled_data=train_data, 161 labeled_data=train_data, 162 labeled_data=train_data, 163 labeled_data=train_data, 164 labeled_data=train_data, 165 labeled_data=train_data, 166 labeled_data=train_data, 167 labeled_data=train_data, 168 labeled_data=train_data, 169 labeled_data=train_data, 170 labeled_data=train_data, 171 labeled_data=train_data, 172 labeled_data=train_data, 173 labeled_data=train_data, 174 labeled_data=train_data, 175 labeled_data=train_data, 176 labeled_data=train_data, 177 labeled_data=train_data, 178 labeled_data=train_data, 179 labeled_data=train_data, 180 labeled_data=train_data, 181 labeled_data=train_data, 182 labeled_data=train_data, 183 labeled_data=train_data, 184 labeled_data=train_data, 185 labeled_data=train_data, 186 labeled_data=train_data, 187 labeled_data=train_data, 188 labeled_data=train_data, 189 labeled_data=train_data, 190 labeled_data=train_data, 191 labeled_data=train_data, 192 labeled_data=train_data, 193 labeled_data=train_data, 194 labeled_data=train_data, 195 labeled_data=train_data, 196 labeled_data=train_data, 197 labeled_data=train_data, 198 labeled_data=train_data, 199 labeled_data=train_data, 200 labeled_data=train_data, 201 labeled_data=train_data, 202 labeled_data=train_data, 203 labeled_data=train_data, 204 labeled_data=train_data, 205 labeled_data=train_data, 206 labeled_data=train_data, 207 labeled_data=train_data, 208 labeled_data=train_data, 209 labeled_data=train_data, 210 labeled_data=train_data, 211 labeled_data=train_data, 212 labeled_data=train_data, 213 labeled_data=train_data, 214 labeled_data=train_data, 215 labeled_data=train_data, 216 labeled_data=train_data, 217 labeled_data=train_data, 218 labeled_data=train_data, 219 labeled_data=train_data, 220 labeled_data=train_data, 221 labeled_data=train_data, 222 labeled_data=train_data, 223 labeled_data=train_data, 224 labeled_data=train_data, 225 labeled_data=train_data, 226 labeled_data=train_data, 227 labeled_data=train_data, 228 labeled_data=train_data, 229 labeled_data=train_data, 230 labeled_data=train_data, 231 labeled_data=train_data, 232 labeled_data=train_data, 233 labeled_data=train_data, 234 labeled_data=train_data, 235 labeled_data=train_data, 236 labeled_data=train_data, 237 labeled_data=train_data, 238 labeled_data=train_data, 239 labeled_data=train_data, 240 labeled_data=train_data, 241 labeled_data=train_data, 242 labeled_data=train_data, 243 labeled_data=train_data, 244 labeled_data=train_data, 245 labeled_data=train_data, 246 labeled_data=train_data, 247 labeled_data=train_data, 248 labeled_data=train_data, 249 labeled_data=train_data, 250 labeled_data=train_data, 251 labeled_data=train_data, 252 labeled_data=train_data, 253 labeled_data=train_data, 254 labeled_data=train_data, 255 labeled_data=train_data, 256 labeled_data=train_data, 257 labeled_data=train_data, 258 labeled_data=train_data, 259 labeled_data=train_data, 260 labeled_data=train_data, 261 labeled_data=train_data, 262 labeled_data=train_data, 263 labeled_data=train_data, 264 labeled_data=train_data, 265 labeled_data=train_data, 266 labeled_data=train_data, 267 labeled_data=train_data, 268 labeled_data=train_data, 269 labeled_data=train_data, 270 labeled_data=train_data, 271 labeled_data=train_data, 272 labeled_data=train_data, 273 labeled_data=train_data, 274 labeled_data=train_data, 275 labeled_data=train_data, 276 labeled_data=train_data, 277 labeled_data=train_data, 278 labeled_data=train_data, 279 labeled_data=train_data, 280 labeled_data=train_data, 281 labeled_data=train_data, 282 labeled_data=train_data, 283 labeled_data=train_data, 284 labeled_data=train_data, 285 labeled_data=train_data, 286 labeled_data=train_data, 287 labeled_data=train_data, 288 labeled_data=train_data, 289 labeled_data=train_data, 290 labeled_data=train_data, 291 labeled_data=train_data, 292 labeled_data=train_data, 293 labeled_data=train_data, 294 labeled_data=train_data, 295 labeled_data=train_data, 296 labeled_data=train_data, 297 labeled_data=train_data, 298 labeled_data=train_data, 299 labeled_data=train_data, 300 labeled_data=train_data, 301 labeled_data=train_data, 302 labeled_data=train_data, 303 labeled_data=train_data, 304 labeled_data=train_data, 305 labeled_data=train_data, 306 labeled_data=train_data, 307 labeled_data=train_data, 308 labeled_data=train_data, 309 labeled_data=train_data, 310 labeled_data=train_data, 311 labeled_data=train_data, 312 labeled_data=train_data, 313 labeled_data=train_data, 314 labeled_data=train_data, 315 labeled_data=train_data, 316 labeled_data=train_data, 317 labeled_data=train_data, 318 labeled_data=train_data, 319 labeled_data=train_data, 320 labeled_data=train_data, 321 labeled_data=train_data, 322 labeled_data=train_data, 323 labeled_data=train_data, 324 labeled_data=train_data, 325 labeled_data=train_data, 326 labeled_data=train_data, 327 labeled_data=train_data, 328 labeled_data=train_data, 329 labeled_data=train_data, 330 labeled_data=train_data, 331 labeled_data=train_data, 332 labeled_data=train_data, 333 labeled_data=train_data, 334 labeled_data=train_data, 335 labeled_data=train_data, 336 labeled_data=train_data, 337 labeled_data=train_data, 338 labeled_data=train_data, 339 labeled_data=train_data, 340 labeled_data=train_data, 341 labeled_data=train_data, 342 labeled_data=train_data, 343 labeled_data=train_data, 344 labeled_data=train_data, 345 labeled_data=train_data, 346 labeled_data=train_data, 347 labeled_data=train_data, 348 labeled_data=train_data, 349 labeled_data=train_data, 350 labeled_data=train_data, 351 labeled_data=train_data, 352 labeled_data=train_data, 353 labeled_data=train_data, 354 labeled_data=train_data, 355 labeled_data=train_data, 356 labeled_data=train_data, 357 labeled_data=train_data, 358 labeled_data=train_data, 359 labeled_data=train_data, 360 labeled_data=train_data, 361 labeled_data=train_data, 362 labeled_data=train_data, 363 labeled_data=train_data, 364 labeled_data=train_data, 365 labeled_data=train_data, 366 labeled_data=train_data, 367 labeled_data=train_data, 368 labeled_data=train_data, 369 labeled_data=train_data, 370 labeled_data=train_data, 371 labeled_data=train_data, 372 labeled_data=train_data, 373 labeled_data=train_data, 374 labeled_data=train_data, 375 labeled_data=train_data, 376 labeled_data=train_data, 377 labeled_data=train_data, 378 labeled_data=train_data, 379 labeled_data=train_data, 380 labeled_data=train_data, 381 labeled_data=train_data, 382 labeled_data=train_data, 383 labeled_data=train_data, 384 labeled_data=train_data, 385 labeled_data=train_data, 386 labeled_data=train_data, 387 labeled_data=train_data, 388 labeled_data=train_data, 389 labeled_data=train_data, 390 labeled_data=train_data, 391 labeled_data=train_data, 392 labeled_data=train_data, 393 labeled_data=train_data, 394 labeled_data=train_data, 395 labeled_data=train_data, 396 labeled_data=train_data, 397 labeled_data=train_data, 398 labeled_data=train_data, 399 labeled_data=train_data, 400 labeled_data=train_data, 401 labeled_data=train_data, 402 labeled_data=train_data, 403 labeled_data=train_data, 404 labeled_data=train_data, 405 labeled_data=train_data, 406 labeled_data=train_data, 407 labeled_data=train_data, 408 labeled_data=train_data, 409 labeled_data=train_data, 410 labeled_data=train_data, 411 labeled_data=train_data, 412 labeled_data=train_data, 413 labeled_data=train_data, 414 labeled_data=train_data, 415 labeled_data=train_data, 416 labeled_data=train_data, 417 labeled_data=train_data, 418 labeled_data=train_data, 419 labeled_data=train_data, 420 labeled_data=train_data, 421 labeled_data=train_data, 422 labeled_data=train_data, 423 labeled_data=train_data, 424 labeled_data=train_data, 425 labeled_data=train_data, 426 labeled_data=train_data, 427 labeled_data=train_data, 428 labeled_data=train_data, 429 labeled_data=train_data, 430 labeled_data=train_data, 431 labeled_data=train_data, 432 labeled_data=train_data, 433 labeled_data=train_data, 434 labeled_data=train_data, 435 labeled_data=train_data, 436 labeled_data=train_data, 437 labeled_data=train_data, 438 labeled_data=train_data, 439 labeled_data=train_data, 440 labeled_data=train_data, 441 labeled_data=train_data, 442 labeled_data=train_data, 443 labeled_data=train_data, 444 labeled_data=train_data, 445 labeled_data=train_data, 446 labeled_data=train_data, 447 labeled_data=train_data, 448 labeled_data=train_data, 449 labeled_data=train_data, 450 labeled_data=train_data, 451 labeled_data=train_data, 452 labeled_data=train_data, 453 labeled_data=train_data, 454 labeled_data=train_data, 455 labeled_data=train_data, 456 labeled_data=train_data, 457 labeled_data=train_data, 458 labeled_data=train_data, 459 labeled_data=train_data, 460 labeled_data=train_data, 461 labeled_data=train_data, 462 labeled_data=train_data, 463 labeled_data=train_data, 464 labeled_data=train_data, 465 labeled_data=train_data, 466 labeled_data=train_data, 467 labeled_data=train_data, 468 labeled_data=train_data, 469 labeled_data=train_data, 470 labeled_data=train_data, 471 labeled_data=train_data, 472 labeled_data=train_data, 473 labeled_data=train_data, 474 labeled_data=train_data, 475 labeled_data=train_data, 476 labeled_data=train_data, 477 labeled_data=train_data, 478 labeled_data=train_data, 479 labeled_data=train_data, 480 labeled_data=train_data, 481 labeled_data=train_data, 482 labeled_data=train_data, 483 labeled_data=train_data, 484 labeled_data=train_data, 485 labeled_data=train_data, 486 labeled_data=train_data, 487 labeled_data=train_data, 488 labeled_data=train_data, 489 labeled_data=train_data, 490 labeled_data=train_data, 491 labeled_data=train_data, 492 labeled_data=train_data, 493 labeled_data=train_data, 494 labeled_data=train_data, 495 labeled_data=train_data, 496 labeled_data=train_data, 497 labeled_data=train_data, 498