ICL

• Arguments
  • DecoderOnlyEmbedderICLModelArguments
  • DecoderOnlyEmbedderICLDataArguments
• Dataset
  • DecoderOnlyEmbedderICLSameDatasetTrainDataset
    • DecoderOnlyEmbedderICLSameDatasetTrainDataset
    • Methods
      • DecoderOnlyEmbedderICLSameDatasetTrainDataset._create_batch_data()
  • AbsEmbedderSameDatasetCollator
    • AbsEmbedderSameDatasetCollator
• Modeling
  • BiDecoderOnlyEmbedderICLModel
  • Methods
    • BiDecoderOnlyEmbedderICLModel.encode()
    • BiDecoderOnlyEmbedderICLModel.compute_score()
    • BiDecoderOnlyEmbedderICLModel.compute_loss()
    • BiDecoderOnlyEmbedderICLModel.gradient_checkpointing_enable()
    • BiDecoderOnlyEmbedderICLModel.enable_input_require_grads()
    • BiDecoderOnlyEmbedderICLModel.save()
    • BiDecoderOnlyEmbedderICLModel._sentence_embedding()
    • BiDecoderOnlyEmbedderICLModel._compute_similarity()
• Runner
  • DecoderOnlyEmbedderICLRunner
    • DecoderOnlyEmbedderICLRunner.load_tokenizer_and_model()
    • DecoderOnlyEmbedderICLRunner.load_train_dataset()
    • DecoderOnlyEmbedderICLRunner.load_trainer()
    • DecoderOnlyEmbedderICLRunner.run()
• Trainer
  • DecoderOnlyEmbedderICLTrainer