logger.py 文件源码

python
阅读 23 收藏 0 点赞 0 评论 0

项目:pytorch_divcolor 作者: aditya12agd5 项目源码 文件源码
def __init__(self, server, port, outdir):
    self.vis = visdom.Visdom(port=port, server=server)

    titles = ['VAE -- KL Div', 'VAE -- Weighted L2', 'VAE -- L2']
    self.vis_plot_vae = []
    for title in titles:
      self.vis_plot_vae.append(self.vis.line(
        X=np.array([0.], dtype='f'),
        Y=np.array([0.], dtype='f'),
        opts=dict(
          xlabel='Iteration',\
          ylabel='Loss',\
          title=title)))

    self.vis_plot_test_vae = self.vis.line(
        X=np.array([0.], dtype='f'),
        Y=np.array([0.], dtype='f'),
        opts=dict(
          xlabel='Iteration',\
          ylabel='Test Loss',\
          title='VAE Test Loss'))


    self.vis_plot_mdn = []
    titles = ['MDN Loss', 'MDN -- L2']
    for title in titles:
      self.vis_plot_mdn.append(self.vis.line(
        X=np.array([0.], dtype='f'),
        Y=np.array([0.], dtype='f'),
        opts=dict(
          xlabel='Iteration',\
          ylabel='Loss',\
          title=title)))

    self.fp_vae = open('%s/log_vae.txt' % outdir, 'w')
    self.fp_vae.write('Iteration; KLDiv; WeightedL2; L2;\n')
    self.fp_vae.flush()

    self.fp_test_vae = open('%s/log_test_vae.txt' % outdir, 'w')
    self.fp_test_vae.write('Iteration; Loss;\n')
    self.fp_test_vae.flush()

    self.fp_mdn = open('%s/log_mdn.txt' % outdir, 'w')
    self.fp_mdn.write('Iteration; Loss; L2 Loss;\n')
    self.fp_mdn.flush()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号