diff --git a/compose.yaml b/compose.yaml index 10bd9e0..2019cde 100644 --- a/compose.yaml +++ b/compose.yaml @@ -34,13 +34,15 @@ services: container_name: cnn_api environment: - PARALLEL_INFERENCES=4 - - QUEUED_INFERENCES=2 - - TORCH_THREADS=2 + - QUEUED_INFERENCES=4 + - TORCH_THREADS=1 - FACE_MAX_FPS=5.5 - PREDICT_RATE_LIMIT=10 - PREDICT_RATE_WINDOW=60 - REVIEW_RATE_LIMIT=5 - REVIEW_RATE_WINDOW=60 + cpus: "5" + mem_limit: 3g cnn_website: