File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -85,14 +85,14 @@ def _batch_inference(
8585 src_pretranslations = stack .enter_context (self ._translation_file_service .get_source_pretranslations ())
8686 writer = stack .enter_context (self ._translation_file_service .open_target_pretranslation_writer ())
8787 current_inference_step = 0
88- phase_progress (ProgressStatus .from_step (current_inference_step , inference_step_count ))
88+ phase_progress (ProgressStatus .from_step (current_inference_step , inference_step_count , "inference" ))
8989 batch_size = self ._config ["inference_batch_size" ]
9090 for pi_batch in batch (src_pretranslations , batch_size ):
9191 if check_canceled is not None :
9292 check_canceled ()
9393 _translate_batch (engine , pi_batch , writer )
9494 current_inference_step += len (pi_batch )
95- phase_progress (ProgressStatus .from_step (current_inference_step , inference_step_count ))
95+ phase_progress (ProgressStatus .from_step (current_inference_step , inference_step_count , "inference" ))
9696
9797 def _save_model (self ) -> None :
9898 logger .info ("Saving model" )
You can’t perform that action at this time.
0 commit comments