Skip to content
Snippets Groups Projects
Commit 7e74d72e authored by Julien B.'s avatar Julien B.
Browse files

fix(trainer): add debug

parent 472a4acc
Branches
No related tags found
No related merge requests found
...@@ -53,6 +53,7 @@ def process_trainer(): ...@@ -53,6 +53,7 @@ def process_trainer():
finetuned_repo_name=job.job_data['finetuned_repo_name'], finetuned_repo_name=job.job_data['finetuned_repo_name'],
huggingface_token=job.job_data['huggingface_token'], huggingface_token=job.job_data['huggingface_token'],
) )
logger.debug(request)
responses = stub.StartTraining(request) responses = stub.StartTraining(request)
for response in responses: for response in responses:
logger.debug(f"gRPC message : {response.status}") logger.debug(f"gRPC message : {response.status}")
......
...@@ -28,11 +28,12 @@ class TrainerServicer(trainer_pb2_grpc.TrainerServicer): ...@@ -28,11 +28,12 @@ class TrainerServicer(trainer_pb2_grpc.TrainerServicer):
def StartTraining(self, request, context): def StartTraining(self, request, context):
def response_messages(): def response_messages():
print("reception d'un event") print("event received")
global is_busy global is_busy
if not is_busy: if not is_busy:
is_busy = True is_busy = True
print(f"incoming request : {request}")
training_process( training_process(
training_data=request.training_data, training_data=request.training_data,
fondation_model_id=request.fondation_model_id, fondation_model_id=request.fondation_model_id,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment