(sample, step=None)
| 83 | |
| 84 | |
| 85 | def log_image(sample, step=None): |
| 86 | if is_main_process(): |
| 87 | sample = array2grid(sample, to255=True) |
| 88 | wandb.log({f"samples": wandb.Image(sample), "train_step": step}) |
| 89 | |
| 90 | |
| 91 | def array2grid(x, to255=False): |
nothing calls this directly
no test coverage detected