diff --git a/create_seed_checkpoint.sh b/create_seed_checkpoint.sh index 3dfbde71..77185bfc 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 8da1ebda..a4107806 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 f58b089a..02148b84 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=""