diff options
| author | Taesung Park <taesung89@gmail.com> | 2017-11-27 22:07:02 -0800 |
|---|---|---|
| committer | Taesung Park <taesung89@gmail.com> | 2017-11-27 22:07:02 -0800 |
| commit | 1615932f9180a7a9df92f33fbb8749aec432d3d9 (patch) | |
| tree | d4e5d060d3bfd5e5c2e165af95e42e1607e418b2 /options | |
| parent | 1171d57b5db2d997a5ba3739e680af261f3a2e8a (diff) | |
changed default weight initialization to normal
Diffstat (limited to 'options')
| -rw-r--r-- | options/base_options.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/options/base_options.py b/options/base_options.py index 28ca673..13466bf 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -38,7 +38,7 @@ class BaseOptions(): 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.') self.parser.add_argument('--resize_or_crop', type=str, default='resize_and_crop', help='scaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop]') self.parser.add_argument('--no_flip', action='store_true', help='if specified, do not flip the images for data augmentation') - self.parser.add_argument('--init_type', type=str, default='xavier', help='network initialization [normal|xavier|kaiming|orthogonal]') + self.parser.add_argument('--init_type', type=str, default='normal', help='network initialization [normal|xavier|kaiming|orthogonal]') self.initialized = True |