use ml.g4dn.2xlarge for train

2024-05-18 16:14:35 +08:00 · 2024-05-18 16:14:35 +08:00 · 69df5d7a86
parent 94a0b6ea6e
commit 69df5d7a86
1 changed files with 1 additions and 1 deletions
--- a/test/config.py
+++ b/test/config.py
@ -94,7 +94,7 @@ logger.info(f"config.ckpt_message: {ckpt_message}")

 train_instance_type = os.environ.get("TRAIN_INSTANCE_TYPE", "ml.g5.2xlarge")
 if region_name == "ap-southeast-1":
-    train_instance_type = "ml.g4dn.12xlarge"
+    train_instance_type = "ml.g4dn.2xlarge"
 if is_gcr:
    train_instance_type = "ml.g4dn.2xlarge"
 logger.info(f"config.train_instance_type: {train_instance_type}")