常用代码速记

Images

from torchvision import transforms
from PIL import Image

img = transforms.ToPILImage()(img[0])
img.save("img.jpg")

wandb

Upload images

images = wandb.Image(data[0], caption=label[0])
wandb.log({"image": images})

Upload logs

wandb.log(step=epoch + 1,
        data={'train_loss':loss, 'epoch': epoch + 1, 'lr':get_lr(sub_optimizer_1), 
        'val/val_acc': acc, 'test/test_acc': test_acc, 
        })