Finetune

• Embedder
  • AbsArguments
    • AbsRerankerModelArguments
    • AbsRerankerDataArguments
  • AbsDataset
    • AbsEmbedderTrainDataset
      • AbsEmbedderTrainDataset
      • Methods
        • AbsEmbedderTrainDataset._load_dataset()
        • AbsEmbedderTrainDataset._shuffle_text()
    • AbsEmbedderCollator
      • AbsEmbedderCollator
    • AbsEmbedderSameDatasetTrainDataset
      • AbsEmbedderSameDatasetTrainDataset
      • Methods
        • AbsEmbedderSameDatasetTrainDataset.refresh_epoch()
        • AbsEmbedderSameDatasetTrainDataset._load_dataset()
        • AbsEmbedderSameDatasetTrainDataset._get_file_batch_size()
        • AbsEmbedderSameDatasetTrainDataset._get_train_group_size()
        • AbsEmbedderSameDatasetTrainDataset._create_batch_data()
    • AbsEmbedderSameDatasetCollator
      • AbsEmbedderSameDatasetCollator
    • EmbedderTrainerCallbackForDataRefresh
      • EmbedderTrainerCallbackForDataRefresh
      • Methods
        • EmbedderTrainerCallbackForDataRefresh.on_epoch_end()
  • AbsModeling
    • AbsEmbedderModel
      • AbsEmbedderModel
      • Methods
        • AbsEmbedderModel.encode()
        • AbsEmbedderModel.compute_loss()
        • AbsEmbedderModel.compute_score()
        • AbsEmbedderModel.save()
        • AbsEmbedderModel.get_local_score()
        • AbsEmbedderModel.compute_local_score()
        • AbsEmbedderModel.forward()
        • AbsEmbedderModel.distill_loss()
        • AbsEmbedderModel._compute_no_in_batch_neg_loss()
        • AbsEmbedderModel._compute_in_batch_neg_loss()
        • AbsEmbedderModel._compute_cross_device_neg_loss()
        • AbsEmbedderModel._dist_gather_tensor()
    • EmbedderOutput
      • EmbedderOutput
  • AbsTrainer
    • AbsEmbedderTrainer
      • AbsEmbedderTrainer
      • Methods
        • AbsEmbedderTrainer.compute_loss()
  • AbsRunner
    • AbsEmbedderTrainer
      • AbsEmbedderRunner
      • Methods
        • AbsEmbedderRunner.load_tokenizer_and_model()
        • AbsEmbedderRunner.load_trainer()
        • AbsEmbedderRunner.load_train_dataset()
        • AbsEmbedderRunner.load_data_collator()
        • AbsEmbedderRunner.run()
• Reranker
  • AbsArguments
    • AbsEmbedderModelArguments
    • AbsEmbedderDataArguments
  • AbsDataset
    • AbsRerankerTrainDataset
      • AbsRerankerTrainDataset
      • Methods
        • AbsRerankerTrainDataset.create_one_example()
        • AbsRerankerTrainDataset._load_dataset()
        • AbsRerankerTrainDataset._shuffle_text()
    • AbsRerankerCollator
      • AbsRerankerCollator
    • AbsLLMRerankerTrainDataset
      • AbsLLMRerankerTrainDataset
    • AbsLLMRerankerCollator
      • AbsLLMRerankerCollator
  • AbsModeling
    • AbsRerankerModel
      • AbsRerankerModel
      • Methods
        • AbsRerankerModel.encode()
        • AbsRerankerModel.gradient_checkpointing_enable()
        • AbsRerankerModel.enable_input_require_grads()
        • AbsRerankerModel.forward()
        • AbsRerankerModel.compute_loss()
        • AbsRerankerModel.save()
        • AbsRerankerModel.save_pretrained()
    • RerankerOutput
      • RerankerOutput
  • AbsTrainer
    • AbsRerankerTrainer
      • AbsRerankerTrainer
      • Methods
        • AbsRerankerTrainer.compute_loss()
  • AbsRunner
    • AbsRerankerTrainer
      • AbsRerankerRunner
      • Methods
        • AbsRerankerRunner.load_tokenizer_and_model()
        • AbsRerankerRunner.load_trainer()
        • AbsRerankerRunner.load_train_dataset()
        • AbsRerankerRunner.load_data_collator()
        • AbsRerankerRunner.run()