Lucas Georges Gabriel Charpentier Pacheco commited on
Commit ·
0dffe6b
1
Parent(s): c9e634c
Fixing size of AoA
Browse files- src/display/utils.py +1 -1
src/display/utils.py
CHANGED
|
@@ -367,7 +367,7 @@ NUM_EXPECTED_EXAMPLES = {
|
|
| 367 |
}
|
| 368 |
}
|
| 369 |
|
| 370 |
-
AOA_SIZE =
|
| 371 |
|
| 372 |
NUM_CHECKPOINTS_10M = 19
|
| 373 |
CHECKPOINT_WORD_COUNTS_10M = [f'{i}M' for i in range(1, 10)] + [f'{i*10}M' for i in range(1, 11)]
|
|
|
|
| 367 |
}
|
| 368 |
}
|
| 369 |
|
| 370 |
+
AOA_SIZE = 8005
|
| 371 |
|
| 372 |
NUM_CHECKPOINTS_10M = 19
|
| 373 |
CHECKPOINT_WORD_COUNTS_10M = [f'{i}M' for i in range(1, 10)] + [f'{i*10}M' for i in range(1, 11)]
|