search

Dashvvood / ima206_project

Mozilla Public License 2.0
1 stars 0 forks source link

Wandb log images #2

Open Dashvvood opened 4 weeks ago

Dashvvood commented 4 weeks ago

发现好几种写法:

  1. link

    sample_imgs = x[:6]
grid = torchvision.utils.make_grid(sample_imgs)
self.logger.experiment.add_image('example_images', grid, 0)
  2. trainer.logger.log_image(key="val_epoch_cm", images=[cm_img], step=pl_module.current_epoch)
Dashvvood commented 4 weeks ago

而且输入参数step=pl_module.current_epoch 最后还是显示的step， 很迷惑