From 7e74d72e163975b3a968254d4b7c5a72fefcb685 Mon Sep 17 00:00:00 2001
From: "Julien B." <xm9q8f80@jlnbrtn.me>
Date: Fri, 23 Aug 2024 10:01:29 +0200
Subject: [PATCH] fix(trainer): add debug
---
api/internal_services/background_worker.py | 1 +
microservices/trainer/trainer.py | 3 ++-
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/api/internal_services/background_worker.py b/api/internal_services/background_worker.py
index 5ce476a..8ed9d8a 100644
--- a/api/internal_services/background_worker.py
+++ b/api/internal_services/background_worker.py
@@ -53,6 +53,7 @@ def process_trainer():
finetuned_repo_name=job.job_data['finetuned_repo_name'],
huggingface_token=job.job_data['huggingface_token'],
)
+ logger.debug(request)
responses = stub.StartTraining(request)
for response in responses:
logger.debug(f"gRPC message : {response.status}")
diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py
index 8016b71..7e7919e 100644
--- a/microservices/trainer/trainer.py
+++ b/microservices/trainer/trainer.py
@@ -28,11 +28,12 @@ class TrainerServicer(trainer_pb2_grpc.TrainerServicer):
def StartTraining(self, request, context):
def response_messages():
- print("reception d'un event")
+ print("event received")
global is_busy
if not is_busy:
is_busy = True
+ print(f"incoming request : {request}")
training_process(
training_data=request.training_data,
fondation_model_id=request.fondation_model_id,
--
GitLab