diff options
Diffstat (limited to 'options/base_options.py')
| -rw-r--r-- | options/base_options.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/options/base_options.py b/options/base_options.py index f3e93c6..575d0d1 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -43,6 +43,12 @@ class BaseOptions(): self.parser.add_argument('--init_type', type=str, default='normal', help='network initialization [normal|xavier|kaiming|orthogonal]') self.parser.add_argument('--center_crop', action='store_true', help='center crop instead of random crop') self.parser.add_argument('--poll_delay', type=float, default=0.01, help='time to wait before checking for the next frame') + self.parser.add_argument( + '--module-name', + type=str, + default='pix2pix', + help='module name... basically says where to look for sequences' + ) self.initialized = True |