diff options
| -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') |