diff --git a/test/config.py b/test/config.py index 55253d10..6f18713a 100644 --- 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}")