summaryrefslogtreecommitdiff
path: root/util/visualizer.py
diff options
context:
space:
mode:
authorGrégoire Payen de La Garanderie <gdlg@hochet.info>2018-02-02 18:44:53 +0000
committerGrégoire Payen de La Garanderie <gdlg@hochet.info>2018-02-02 18:46:14 +0000
commit3d83bfe9e9d2f335b50a28b20e0427a66d365fca (patch)
treed6c0c4ef8d684f3eb2e6f327a7654fbf4ef26cd2 /util/visualizer.py
parent7a5e2cd5f5003e8ca9a0fc3dac14a74b81287881 (diff)
Add --display_server option to choose Visdom server.
Diffstat (limited to 'util/visualizer.py')
-rw-r--r--util/visualizer.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/util/visualizer.py b/util/visualizer.py
index 8bec8df..08eeb75 100644
--- a/util/visualizer.py
+++ b/util/visualizer.py
@@ -18,7 +18,7 @@ class Visualizer():
self.saved = False
if self.display_id > 0:
import visdom
- self.vis = visdom.Visdom(port=opt.display_port)
+ self.vis = visdom.Visdom(server=opt.display_server, port=opt.display_port)
if self.use_html:
self.web_dir = os.path.join(opt.checkpoints_dir, opt.name, 'web')