diff --git a/create_seed_checkpoint.sh b/create_seed_checkpoint.sh index 3dfbde714..77185bfce 100755 --- a/create_seed_checkpoint.sh +++ b/create_seed_checkpoint.sh @@ -18,7 +18,6 @@ set -ex -TRAINER_DIR=${1:-/home/$USER/local/torchtitan} NGPU=1 LOG_RANK=0 CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"} diff --git a/run_llama_train.sh b/run_llama_train.sh index 8da1ebda4..a41078068 100755 --- a/run_llama_train.sh +++ b/run_llama_train.sh @@ -11,7 +11,6 @@ set -ex # e.g. # LOG_RANK=0,1 NGPU=4 ./run_llama_train.sh NGPU=${NGPU:-"8"} -TRAINER_DIR=${TRAINER_DIR:-/home/$USER/local/torchtitan} LOG_RANK=${LOG_RANK:-0} CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"} diff --git a/run_memory_estimation.sh b/run_memory_estimation.sh index f58b089ad..02148b84f 100755 --- a/run_memory_estimation.sh +++ b/run_memory_estimation.sh @@ -12,7 +12,6 @@ set -ex # NGPU=4 ./run_memory_estimation.sh NGPU=${NGPU:-"8"} NNODES=${NNODES:-"1"} -TRAINER_DIR=${TRAINER_DIR:-/home/$USER/local/torchtitan} CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"} overrides=""