summaryrefslogtreecommitdiff
path: root/options
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 /options
parent7a5e2cd5f5003e8ca9a0fc3dac14a74b81287881 (diff)
Add --display_server option to choose Visdom server.
Diffstat (limited to 'options')
-rw-r--r--options/base_options.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/options/base_options.py b/options/base_options.py
index 13466bf..1b5ebaf 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"), help='Maximum number of samples allowed per dataset. If the dataset directory contains more than max_dataset_size, only a subset is loaded.')