Skip to content

Commit 6c448d7

Browse files
committed
rm print(self.model) in trainer.py for reranker
1 parent a08f352 commit 6c448d7

1 file changed

Lines changed: 0 additions & 1 deletion

File tree

  • FlagEmbedding/finetune/reranker/encoder_only/base

FlagEmbedding/finetune/reranker/encoder_only/base/trainer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ def _save(self, output_dir: Optional[str] = None, state_dict=None):
2929
if not hasattr(self.model, 'save_pretrained'):
3030
raise NotImplementedError(f'MODEL {self.model.__class__.__name__} ' f'does not support save_pretrained interface')
3131
else:
32-
print(self.model)
3332
self.model.save_pretrained(output_dir)
3433
if self.tokenizer is not None and self.is_world_process_zero():
3534
self.tokenizer.save_pretrained(output_dir)

0 commit comments

Comments
 (0)