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 2840def..c79dfa2 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -143,7 +143,7 @@ def parse_args(): | |||
| 143 | parser.add_argument( | 143 | parser.add_argument( |
| 144 | "--num_buckets", | 144 | "--num_buckets", |
| 145 | type=int, | 145 | type=int, |
| 146 | default=0, | 146 | default=4, |
| 147 | help="Number of aspect ratio buckets in either direction.", | 147 | help="Number of aspect ratio buckets in either direction.", |
| 148 | ) | 148 | ) |
| 149 | parser.add_argument( | 149 | parser.add_argument( |