Source code for draugr.torch_utilities.writers.visdom.visdom_writer

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import numpy
import visdom

from draugr.writers.writer import Writer

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 9/14/19
           """

__all__ = ["VisdomWriter"]

# Visualisation
USE_VISDOM = False
START_VISDOM_SERVER = False
VISDOM_SERVER = "http://localhost"
if not START_VISDOM_SERVER:
    # noinspection PyRedeclaration
    VISDOM_SERVER = "http://visdom.ml"


[docs]class VisdomWriter(Writer): """description"""
[docs] def __init__(self): super().__init__()
def _scalar(self, tag: str, value: float, step: int): self.server.line( Y=numpy.array([value]), X=numpy.array([step]), win=tag, update="append" ) def _close(self, exc_type=None, exc_val=None, exc_tb=None): # self.server.close() # close a window by id del self.server def _open(self): self.server = visdom.Visdom(server=VISDOM_SERVER, port=8097)