From 5c7f0d36a9e4410014b55f087327ad5b76353c6a Mon Sep 17 00:00:00 2001 From: Heiru Wu Date: Sun, 26 Nov 2023 03:20:31 +0800 Subject: [PATCH] chore: update resource reservation --- yolov7/1/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/yolov7/1/model.py b/yolov7/1/model.py index a829b48..0c0bccf 100644 --- a/yolov7/1/model.py +++ b/yolov7/1/model.py @@ -446,8 +446,8 @@ def undeploy_model(model_name: str): ray.init(address=model_config.ray_addr) - model_config.ray_actor_options["num_cpus"] = 1 - model_config.ray_actor_options["num_gpus"] = 0.2 + model_config.ray_actor_options["num_cpus"] = 0.4 + model_config.ray_actor_options["num_gpus"] = 0.1 if func == "deploy": deploy_model(model_config=model_config)