(batch)
| 351 | |
| 352 | |
| 353 | def collate_fn(batch): |
| 354 | # import pdb; pdb.set_trace() |
| 355 | batch = list(zip(*batch)) |
| 356 | batch[0] = nested_tensor_from_tensor_list(batch[0]) |
| 357 | return tuple(batch) |
| 358 | |
| 359 | |
| 360 | def _max_by_axis(the_list): |
nothing calls this directly
no test coverage detected