PyTorch深度學習快速入門教程（絕對通俗易懂?。拘⊥炼选?/h1>
2022-06-29 10:17 作者:偏執(zhí)的小包菜 | 我要投稿
可以使用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()
標簽：
