From 8fa11f0f55a6f2d2e5bf460098546e600cc3058e Mon Sep 17 00:00:00 2001 From: Will Constable Date: Sun, 14 Jul 2024 21:17:04 -0700 Subject: [PATCH] [Cleanup] Remove unused TRAINER_DIR This argument seems to be left over from older times- it is not used anywhere in the codebase. ghstack-source-id: abbcf82ed4d1b8fbb71c6a6b48acbc1296dbec64 Pull Request resolved: https://github.com/pytorch/torchtitan/pull/456 --- create_seed_checkpoint.sh | 1 - run_llama_train.sh | 1 - run_memory_estimation.sh | 1 - 3 files changed, 3 deletions(-) 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=""