From 2078c5950074aa12923f00c63643af218c13b29a Mon Sep 17 00:00:00 2001 From: dimapihtar Date: Tue, 19 Sep 2023 04:26:25 -0700 Subject: [PATCH] change container to 23.08.02 Signed-off-by: dimapihtar --- README.md | 4 ++-- auto_configurator/conf/config.yaml | 2 +- auto_configurator/tests/config_tests/test_main_config.py | 2 +- csp_tools/aws/Dockerfile | 2 +- launcher_scripts/conf/config.yaml | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 455c0be7d6..6ddb16e2b7 100755 --- a/README.md +++ b/README.md @@ -1976,7 +1976,7 @@ launcher_scripts_path: ${auto_configurator_path}/../launcher_scripts fastertransformer_path: ${auto_configurator_path}/../FasterTransformer base_results_dir: ${auto_configurator_path}/results data_dir: ${launcher_scripts_path}/data -training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01 +training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02 container_mounts: - null wandb: # Weights and Biases (W&B) logging. @@ -5400,7 +5400,7 @@ VALID_DATA_PATH=/path/to/val_actor TEST_DATA_PATH=/path/to/test_actor NEMO_RLHF_DIR=/opt/nemo-rlhf -CONTAINER="nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01" +CONTAINER="nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02" mkdir -p $OUTPUT_DIR diff --git a/auto_configurator/conf/config.yaml b/auto_configurator/conf/config.yaml index d28e5060b6..3f12c653ed 100644 --- a/auto_configurator/conf/config.yaml +++ b/auto_configurator/conf/config.yaml @@ -18,7 +18,7 @@ launcher_scripts_path: ${auto_configurator_path}/../launcher_scripts base_results_dir: ${auto_configurator_path}/results data_dir: ${launcher_scripts_path}/data -training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01 +training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02 container_mounts: - null diff --git a/auto_configurator/tests/config_tests/test_main_config.py b/auto_configurator/tests/config_tests/test_main_config.py index b989c57cfd..29d53c7fa6 100644 --- a/auto_configurator/tests/config_tests/test_main_config.py +++ b/auto_configurator/tests/config_tests/test_main_config.py @@ -26,7 +26,7 @@ def test_config(self): base_results_dir: ${auto_configurator_path}/results data_dir: ${launcher_scripts_path}/data - training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01 + training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02 container_mounts: - null diff --git a/csp_tools/aws/Dockerfile b/csp_tools/aws/Dockerfile index bf62aa76a0..424a868b3c 100755 --- a/csp_tools/aws/Dockerfile +++ b/csp_tools/aws/Dockerfile @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -FROM nvcr.io/ea-bignlp/nemofw-training:23.07-py3 +FROM nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02 ARG EFA_INSTALLER_VERSION=latest ARG AWS_OFI_NCCL_VERSION=1.4.0-aws diff --git a/launcher_scripts/conf/config.yaml b/launcher_scripts/conf/config.yaml index 0e39bbe0af..8458294910 100755 --- a/launcher_scripts/conf/config.yaml +++ b/launcher_scripts/conf/config.yaml @@ -39,7 +39,7 @@ data_dir: ${launcher_scripts_path}/data # Location to store and read the data. base_results_dir: ${launcher_scripts_path}/results # Location to store the results, checkpoints and logs. container_mounts: # List of additional paths to mount to container. They will be mounted to same path. - null -container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01 +container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02 wandb_api_key_file: null # File where the w&B api key is stored. Key must be on the first line.