README.md

VisdomPortal

Portal for Visdom and PyTorch

Install

pip install git+https://github.com/automan000/VisdomPortal

Requirements

• PyTorch-0.4.0
• Visdom
• Numpy

Usage

vis = VisdomPortal(env_name='test')

parameters:

• env_name=some string
• image_norm=[(0.5, 0.5, 0.5), (0.5, 0.5, 0.5)] # similar to torchvision.transforms.normalize

Draw curve

vis.draw_curve(value=loss, step=index, title='loss')

parameters:

• value: torch.autograd.variable.Variable, int, float, or np.ndarray
• step: int or numpy

Draw multiple images

vis.draw_images(value=images, title='Images', unnormalize=True)

parameters:

• value: 4-D torch.autograd.variable.Variable, torch._TensorBase, or np.ndarray
• title: string
• unnormalize: if un-normalize input values