PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】

可以使用for step,item enumerate(dataloader)来使程序更加的Pythonic哦!

from torch.utils.tensorboard import SummaryWriter

test_loader = DataLoader(dataset=test_set, batch_size=64,

                         shuffle=True, drop_last=False)

for times in range(2):                                      # shuffle=True则两次结果不一样

    writer = SummaryWriter('../logs/log_Dataloader')

    for step, item in enumerate(test_loader):

        imgs_arr, kinds_arr = item

        writer.add_images(tag='shuffle=True'+str(times),

                          img_tensor=imgs_arr, global_step=step+1, dataformats="NCHW")

    writer.close()