diff options
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 55daa35..e272b5d 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -83,7 +83,7 @@ def parse_args(): | |||
| 83 | parser.add_argument( | 83 | parser.add_argument( |
| 84 | "--num_class_images", | 84 | "--num_class_images", |
| 85 | type=int, | 85 | type=int, |
| 86 | default=400, | 86 | default=1, |
| 87 | help="How many class images to generate." | 87 | help="How many class images to generate." |
| 88 | ) | 88 | ) |
| 89 | parser.add_argument( | 89 | parser.add_argument( |