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