(self, pl_module, images, batch_idx, split)
| 512 | |
| 513 | @rank_zero_only |
| 514 | def _testtube(self, pl_module, images, batch_idx, split): |
| 515 | for k in images: |
| 516 | grid = torchvision.utils.make_grid(images[k]) |
| 517 | grid = (grid + 1.0) / 2.0 # -1,1 -> 0,1; c,h,w |
| 518 | |
| 519 | tag = f"{split}/{k}" |
| 520 | pl_module.logger.experiment.add_image( |
| 521 | tag, grid, |
| 522 | global_step=pl_module.global_step) |
| 523 | |
| 524 | @rank_zero_only |
| 525 | def log_local(self, save_dir, split, images, |
nothing calls this directly
no outgoing calls
no test coverage detected