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()