def __init__(self, envs=None, interval=[(1, 'epoch')]):
super(VisdomSaver, self).__init__(interval)
self.envs = envs
self.viz = visdom.Visdom()
微信
公众号
扫码关注公众号