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()
评论列表
文章目录