(ios)
| 363 | finish_checked = False |
| 364 | |
| 365 | def finish_callback(ios): |
| 366 | nonlocal finish_checked |
| 367 | finish_checked = True |
| 368 | assert len(ios) == 1 |
| 369 | for key in ios: |
| 370 | io = key |
| 371 | data = ios[key].to_numpy().flatten() |
| 372 | output_data = self.correct_data.flatten() |
| 373 | assert data.size == output_data.size |
| 374 | for i in range(data.size): |
| 375 | assert abs(data[i] - output_data[i]) < 1e-5 |
| 376 | return 0 |
| 377 | |
| 378 | network.set_finish_callback(finish_callback) |
| 379 | self.do_forward(network, 1) |