Config
DatasetConfig
EmbeddingConfig
MetricConfig
ModelConfig
PreprocessorConfig
TrainerConfig
TrainerConfig.batch_size
TrainerConfig.checkpoints_dir
TrainerConfig.config_type
TrainerConfig.dataloader_shuffle
TrainerConfig.device
TrainerConfig.distributed
TrainerConfig.do_evaluate
TrainerConfig.eval_batch_size
TrainerConfig.evaluate_with_generate
TrainerConfig.gradient_accumulation_steps
TrainerConfig.init_weights_from
TrainerConfig.learning_rate
TrainerConfig.log_steps
TrainerConfig.logs_dir
TrainerConfig.lr_scheduler
TrainerConfig.lr_scheduler_kwargs
TrainerConfig.lr_scheduling_steps
TrainerConfig.max_steps
TrainerConfig.metric_for_best_model
TrainerConfig.metrics
TrainerConfig.mixed_precision
TrainerConfig.name
TrainerConfig.num_dataloader_workers
TrainerConfig.num_epochs
TrainerConfig.optimizer
TrainerConfig.output_dir
TrainerConfig.resume_from_checkpoint
TrainerConfig.save_enabled
TrainerConfig.save_freq
TrainerConfig.save_steps
TrainerConfig.seed
TrainerConfig.task
TrainerConfig.use_cpu
TrainerConfig.weight_decay
Backends
ChannelsAxisSide
Color
ConfigType
ExplicitEnum
ImageType
LRSchedulerType
LRSchedulerType.CONSTANT
LRSchedulerType.COSINE_ANEALING
LRSchedulerType.CYCLIC
LRSchedulerType.EXPONENTIAL
LRSchedulerType.LAMBDA
LRSchedulerType.LINEAR
LRSchedulerType.MULTI_STEP
LRSchedulerType.ONE_CYCLE
LRSchedulerType.POLYNOMIAL
LRSchedulerType.REDUCE_ON_PLATEAU
LRSchedulerType.SEQUENTIAL
LRSchedulerType.STEP
LossType
MetricType
OptimizerType
PaddingType
PrecisionType
RegistryType
RepoType
SplitType
TaskType