from tensorboardX import SummaryWriter
import torchvision.utils as vutils
writer = SummaryWriter()
for i, (x, y) in enumerate(dataloader):
xxx
writer.add_scalar('/train/loss', loss)
writer.add_scalar('/train/acc', acc)
writer.add_image('train/inputs', utils.make_grid(x.cpu().data, normalize=True), gloabal_step= xx)
writer.add_text('text', ll, global_step=xx)
writer.close()
reference