File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
torchao/float8/benchmarking Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ LOG_FILE="/tmp/float8_training_log.txt"
13
13
# validate user has specified torchtitan root directory
14
14
if [ -z " ${TORCHTITAN_ROOT} " ]; then
15
15
echo " Error: TORCHTITAN environment variable is not set. Please set it before running this script."
16
- echo " Usage: TORCHTITAN_ROOT=<directory> ./llama3_8b_float8_training_benchmark .sh"
16
+ echo " Usage: TORCHTITAN_ROOT=<directory> ./float8_training_benchmark .sh"
17
17
echo " Optional parameters configurable via environment variables:"
18
18
echo " * RECIPE: rowwise|tensorwise. defaults to tensorwise."
19
19
echo " * BATCH_SIZE: defaults to 1."
Original file line number Diff line number Diff line change 6
6
Usage:
7
7
python parse_torchtitan_logs.py --log-file <log_file_path>
8
8
"""
9
+
9
10
import os
10
11
import re
11
12
import statistics
You can’t perform that action at this time.
0 commit comments