[monitoring tools] Tensorboard
from torch.utils.tensorboard import SummaryWriter import numpy as np img_batch = np.zeros((16, 3, 100, 100)) for i in range(16): img_batch[i, 0] = np.arange(0, 10000).reshape(100, 100)/10000/16*i img_batch[i, 1] = (1 - np.arange(0, 10000).reshape(100, 100)/10000)/16*i writer = SummaryWriter(logs_base_dir) writer.add_images('my_image_batch', img_batch, 0) writer.flush() writer.close() 기본 개념 scala..
더보기
