Base

• Modeling
  • BiEncoderOnlyEmbedderModel
  • Methods
    • BiEncoderOnlyEmbedderModel.encode()
    • BiEncoderOnlyEmbedderModel.compute_score()
    • BiEncoderOnlyEmbedderModel.compute_loss()
    • BiEncoderOnlyEmbedderModel.gradient_checkpointing_enable()
    • BiEncoderOnlyEmbedderModel.enable_input_require_grads()
    • BiEncoderOnlyEmbedderModel.save()
    • BiEncoderOnlyEmbedderModel._sentence_embedding()
    • BiEncoderOnlyEmbedderModel._compute_similarity()
• Runner
  • EncoderOnlyEmbedderRunner
    • EncoderOnlyEmbedderRunner.load_tokenizer_and_model()
    • EncoderOnlyEmbedderRunner.load_trainer()
• Trainer
  • EncoderOnlyEmbedderTrainer