diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2018-03-26 14:58:45 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-03-26 14:58:45 -0400 |
| commit | 3a853ff09c1d3b82a649155d8c576f8d10eb1a65 (patch) | |
| tree | 6ca34f0a993cb94a6ad47e081454c8220e480555 | |
| parent | 6718e13d5ded5f630cba3a8eab7e29fc7a7455e2 (diff) | |
| parent | 3d83bfe9e9d2f335b50a28b20e0427a66d365fca (diff) | |
Merge pull request #201 from gdlg/display_server
Add --display_server option to choose Visdom server.
| -rw-r--r-- | options/base_options.py | 1 | ||||
| -rw-r--r-- | util/visualizer.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/options/base_options.py b/options/base_options.py index ce58548..5bdd85e 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -33,6 +33,7 @@ class BaseOptions(): self.parser.add_argument('--serial_batches', action='store_true', help='if true, takes images in order to make batches, otherwise takes them randomly') self.parser.add_argument('--display_winsize', type=int, default=256, help='display window size') self.parser.add_argument('--display_id', type=int, default=1, help='window id of the web display') + self.parser.add_argument('--display_server', type=str, default="http://localhost", help='visdom server of the web display') self.parser.add_argument('--display_port', type=int, default=8097, help='visdom port of the web display') self.parser.add_argument('--no_dropout', action='store_true', help='no dropout for the generator') self.parser.add_argument('--max_dataset_size', type=int, default=float("inf"), diff --git a/util/visualizer.py b/util/visualizer.py index fd8140d..a98b512 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') |