Base

• Model
  • BaseModel
    • BaseModel
      • BaseModel.device
      • BaseModel.dtype
      • BaseModel.evaluate()
      • BaseModel.get_backend()
      • BaseModel.get_max_seq_length()
      • BaseModel.get_model_kwargs()
      • BaseModel.gradient_checkpointing_enable()
      • BaseModel.is_singular_input()
      • BaseModel.max_seq_length
      • BaseModel.modalities
      • BaseModel.model_card_data_class
      • BaseModel.preprocess()
      • BaseModel.processor
      • BaseModel.push_to_hub()
      • BaseModel.save_pretrained()
      • BaseModel.save_to_hub()
      • BaseModel.start_multi_process_pool()
      • BaseModel.stop_multi_process_pool()
      • BaseModel.supports()
      • BaseModel.tokenize()
      • BaseModel.tokenizer
      • BaseModel.transformers_model
  • BaseModelCardData
    • BaseModelCardData
• Trainer
  • BaseTrainer
    • BaseTrainer
      • BaseTrainer.add_callback()
      • BaseTrainer.add_dataset_name_transform()
      • BaseTrainer.add_model_card_callback()
      • BaseTrainer.compute_loss()
      • BaseTrainer.create_model_card()
      • BaseTrainer.create_optimizer()
      • BaseTrainer.create_optimizer_and_scheduler()
      • BaseTrainer.create_scheduler()
      • BaseTrainer.data_collator_class
      • BaseTrainer.evaluate()
      • BaseTrainer.get_batch_sampler()
      • BaseTrainer.get_data_collator()
      • BaseTrainer.get_eval_dataloader()
      • BaseTrainer.get_learning_rates()
      • BaseTrainer.get_multi_dataset_batch_sampler()
      • BaseTrainer.get_num_trainable_parameters()
      • BaseTrainer.get_optimizer_group()
      • BaseTrainer.get_test_dataloader()
      • BaseTrainer.get_train_dataloader()
      • BaseTrainer.hyperparameter_search()
      • BaseTrainer.is_local_process_zero()
      • BaseTrainer.is_world_process_zero()
      • BaseTrainer.log()
      • BaseTrainer.model_card_callback_class
      • BaseTrainer.model_card_data_class
      • BaseTrainer.model_class
      • BaseTrainer.pop_callback()
      • BaseTrainer.preprocess_dataset()
      • BaseTrainer.propagate_args_to_deepspeed()
      • BaseTrainer.push_to_hub()
      • BaseTrainer.remove_callback()
      • BaseTrainer.save_model()
      • BaseTrainer.set_initial_training_values()
      • BaseTrainer.should_dataset_name_column_be_added()
      • BaseTrainer.train()
      • BaseTrainer.training_args_class
• Training Arguments
  • BaseTrainingArguments
    • BaseTrainingArguments
      • BaseTrainingArguments.ddp_timeout_delta
      • BaseTrainingArguments.device
      • BaseTrainingArguments.eval_batch_size
      • BaseTrainingArguments.get_process_log_level()
      • BaseTrainingArguments.get_warmup_steps()
      • BaseTrainingArguments.local_process_index
      • BaseTrainingArguments.main_process_first()
      • BaseTrainingArguments.n_gpu
      • BaseTrainingArguments.parallel_mode
      • BaseTrainingArguments.place_model_on_device
      • BaseTrainingArguments.process_index
      • BaseTrainingArguments.set_dataloader()
      • BaseTrainingArguments.set_evaluate()
      • BaseTrainingArguments.set_logging()
      • BaseTrainingArguments.set_lr_scheduler()
      • BaseTrainingArguments.set_optimizer()
      • BaseTrainingArguments.set_push_to_hub()
      • BaseTrainingArguments.set_save()
      • BaseTrainingArguments.set_testing()
      • BaseTrainingArguments.set_training()
      • BaseTrainingArguments.should_log
      • BaseTrainingArguments.should_save
      • BaseTrainingArguments.to_dict()
      • BaseTrainingArguments.to_json_string()
      • BaseTrainingArguments.to_sanitized_dict()
      • BaseTrainingArguments.train_batch_size
      • BaseTrainingArguments.world_size
• Samplers
  • BatchSamplers
    • BatchSamplers
    • DefaultBatchSampler
    • NoDuplicatesBatchSampler
    • GroupByLabelBatchSampler
  • MultiDatasetBatchSamplers
    • MultiDatasetBatchSamplers
    • MultiDatasetDefaultBatchSampler
    • RoundRobinBatchSampler
    • ProportionalBatchSampler
• Modules
  • Common Modules
    • Transformer
      • Transformer.get_embedding_dimension()
      • Transformer.max_seq_length
      • Transformer.modalities
      • Transformer.preprocess()
    • Dense
    • Router
      • Router.for_query_document()
  • Base Modules
    • Module
      • Module.config_file_name
      • Module.config_keys
      • Module.forward()
      • Module.get_config_dict()
      • Module.load()
      • Module.load_config()
      • Module.load_dir_path()
      • Module.load_file_path()
      • Module.load_torch_weights()
      • Module.save()
      • Module.save_config()
      • Module.save_in_root
      • Module.save_torch_weights()
    • InputModule
      • InputModule.modalities
      • InputModule.preprocess()
      • InputModule.save_in_root
      • InputModule.save_tokenizer()
      • InputModule.tokenize()
      • InputModule.tokenizer
• Evaluation
  • BaseEvaluator
  • SequentialEvaluator