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():
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}")
......
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment