索引 A | B | C | D | E | F | G | H | I | L | M | O | P | Q | R | S | T | U | W | Z A AccuracyMetric(collie.metrics 中的类) Adan(collie.optim 中的类) apply_to_collection()(在 collie.utils 模块中) B barrier()(Env 静态方法) base_model_prefix(CollieModelForCausalLM 属性) base_model_prefix(InternLMForCausalLM 属性) base_model_prefix(LlamaForCausalLM 属性) base_model_prefix(Moss003MoonForCausalLM 属性) base_model_prefix(MossForCausalLM 属性) BaseMetric(collie.metrics 中的类) BaseMonitor(collie.utils 中的类) BaseProvider(collie.utils 中的类) BleuMetric(collie.metrics 中的类) broadcast_tensor()(在 collie.utils 模块中) C callback_manager(Trainer 属性) callback_name(Callback 属性) Callback(collie.callbacks 中的类) can_generate() (PipelineGenerationMixin 方法) ChatGLMForCausalLM(collie.models 中的类) CheckpointCallback(collie.callbacks 中的类) checkpointing(CollieConfig 属性) ClassifyFPreRecMetric(collie.metrics 中的类) clean_cache() (CollieModelForCausalLM 方法) collate_fn() (ColliePadder 方法) CollieBatchSampler(collie.data 中的类) CollieConfig(collie.config 中的类) CollieDataLoader(collie.data 中的类) CollieDatasetForClassification(collie.data 中的类) CollieDatasetForTraining(collie.data 中的类) CollieModelForCausalLM(collie.models 中的类) ColliePadder(collie.utils 中的类) ColumnParallelLinearWithoutBias(collie.module 中的类) ColumnParallelLMHead(collie.module 中的类) concat_tensor()(在 collie.utils 模块中) construct() (BaseMetric 方法) D DashProvider(collie.utils 中的类) dataloader_num_workers(CollieConfig 属性) DecodeMetric(collie.metrics 中的类) dict_as_params()(在 collie.utils 模块中) dp_group(Env 属性) dp_rank(Env 属性) dp_size(CollieConfig 属性) dp_size(Env 属性) dropout(CollieConfig 属性) ds_config(CollieConfig 属性) E enable_input_require_grads() (CollieModelForCausalLM 方法) Env(collie.utils.dist_utils 中的类) eval() (Trainer 方法) eval_batch_size(CollieConfig 属性) eval_fn()(EvaluatorForClassfication 静态方法) eval_fn()(EvaluatorForGeneration 静态方法) eval_fn()(EvaluatorForPerplexity 静态方法) eval_forward() (PipelineGenerationMixin 方法) eval_per_n_epochs(CollieConfig 属性) eval_per_n_steps(CollieConfig 属性) EvalMonitor(collie.utils 中的类) EvaluatorForClassfication(collie.controller 中的类) EvaluatorForGeneration(collie.controller 中的类) EvaluatorForPerplexity(collie.controller 中的类) F FileIODriver(collie.driver 中的类) find_tensors()(在 collie.utils 模块中) forward() (GPTLMLoss 方法) from_config()(CollieModelForCausalLM 类方法) from_pretrained()(CollieConfig 类方法) from_pretrained()(CollieModelForCausalLM 类方法) fuse_update() (Lomo 方法) fuse_update_zero3() (Lomo 方法) fused_backward() (Lomo 方法) G gather() (BaseMetric 方法) generate() (CollieModelForCausalLM 方法) generate() (PipelineGenerationMixin 方法) generate_forward() (PipelineGenerationMixin 方法) get_data() (BaseProvider 方法) get_feedback() (BaseProvider 方法) get_metric() (AccuracyMetric 方法) get_metric() (ClassifyFPreRecMetric 方法) get_metric() (DecodeMetric 方法) global_batch_idx(Trainer 属性) GPTLMLoss(collie.module 中的类) grad_norm() (Lomo 方法) gradient_accumulation_steps(CollieConfig 属性) GradioProvider(collie.utils 中的类) H HasMonitorCallback(collie.callbacks 中的类) I init_method(CollieConfig 属性) init_state_dict() (Trainer 方法) InternLMForCausalLM(collie.models 中的类) IODriver(collie.driver 中的类) is_first_stage(Env 属性) is_last_stage(Env 属性) is_pipeline(Env 属性) is_zero3_enabled()(在 collie.utils 模块中) L Lion(collie.optim 中的类) LlamaForCausalLM(collie.models 中的类) load_checkpoint() (Trainer 方法) load_parallel_state_dict()(ChatGLMForCausalLM 静态方法) load_parallel_state_dict()(CollieModelForCausalLM 静态方法) load_parallel_state_dict()(InternLMForCausalLM 静态方法) load_parallel_state_dict()(LlamaForCausalLM 静态方法) load_parallel_state_dict()(Moss003MoonForCausalLM 静态方法) load_parallel_state_dict()(MossForCausalLM 静态方法) load_peft() (Trainer 方法) load_state_dict() (Trainer 方法) LoadBestModelCallback(collie.callbacks 中的类) local_rank(Env 属性) Lomo(collie.optim 中的类) LossMonitor(collie.utils 中的类) low_cpu_mem_usage(CollieConfig 属性) LRMonitor(collie.utils 中的类) M main_input_name(CollieModelForCausalLM 属性) MemoryMonitor(collie.utils 中的类) model_config(CollieConfig 属性) Moss003MoonForCausalLM(collie.models 中的类) MossForCausalLM(collie.models 中的类) mp_rank(Env 属性) O on_after_trainer_initialized() (Callback 方法) on_after_trainer_initialized() (HasMonitorCallback 方法) on_evaluate_begin() (Callback 方法) on_evaluate_end() (Callback 方法) on_load_checkpoint() (Callback 方法) on_load_model() (Callback 方法) on_save_checkpoint() (Callback 方法) on_save_model() (Callback 方法) on_setup_parallel_model() (Callback 方法) on_train_batch_begin() (Callback 方法) on_train_batch_end() (Callback 方法) on_train_begin() (Callback 方法) on_train_end() (Callback 方法) on_train_epoch_begin() (Callback 方法) on_train_epoch_end() (Callback 方法) P peft_config(CollieConfig 属性) PetrelIODriver(collie.driver 中的类) pipeline_layers()(ChatGLMForCausalLM 类方法) pipeline_layers()(CollieModelForCausalLM 类方法) pipeline_layers()(InternLMForCausalLM 类方法) pipeline_layers()(LlamaForCausalLM 类方法) pipeline_layers()(MossForCausalLM 类方法) pipeline_layers_idx(Env 属性) pipeline_parts(Env 属性) PipelineGenerationMixin(collie.module 中的类) pp_group(Env 属性) pp_partition_method(CollieConfig 属性) pp_rank(Env 属性) pp_size(CollieConfig 属性) pp_size(Env 属性) PPLMetric(collie.metrics 中的类) prepare_inputs() (CollieModelForCausalLM 方法) prepare_inputs_for_generation() (CollieModelForCausalLM 方法) progress(collie.utils 中的类) provider_handler() (BaseProvider 方法) put_feedback() (BaseProvider 方法) Q quantization_config(CollieConfig 属性) R rank(Env 属性) reset() (AccuracyMetric 方法) reset() (BaseMetric 方法) reset() (ClassifyFPreRecMetric 方法) reset() (progress 方法) RowParallelLinearWithoutBias(collie.module 中的类) S save_checkpoint() (Trainer 方法) save_model() (Trainer 方法) save_parallel_state_dict()(ChatGLMForCausalLM 静态方法) save_parallel_state_dict()(CollieModelForCausalLM 静态方法) save_parallel_state_dict()(InternLMForCausalLM 静态方法) save_parallel_state_dict()(LlamaForCausalLM 静态方法) save_parallel_state_dict()(Moss003MoonForCausalLM 静态方法) save_parallel_state_dict()(MossForCausalLM 静态方法) save_peft() (Trainer 方法) save_pretrained() (CollieConfig 方法) seed(CollieConfig 属性) seed(Env 属性) set_cache() (CollieModelForCausalLM 方法) set_description() (progress 方法) set_engine() (PipelineGenerationMixin 方法) set_post_desc() (progress 方法) set_postfix() (progress 方法) set_seed()(在 collie.utils 模块中) setup_distribution()(在 collie.utils 模块中) setup_ds_engine()(在 collie.utils 模块中) setup_parallel_model() (Trainer 方法) SophiaG(collie.optim 中的类) start_provider() (BaseProvider 方法) state_dict() (Trainer 方法) StepTimeMonitor(collie.utils 中的类) T TGSMonitor(collie.utils 中的类) tp_group(Env 属性) tp_rank(Env 属性) tp_size(CollieConfig 属性) tp_size(Env 属性) train() (Trainer 方法) train_epochs(CollieConfig 属性) train_fn()(Trainer 静态方法) train_forward() (PipelineGenerationMixin 方法) train_micro_batch_size(CollieConfig 属性) Trainer(collie.controller 中的类) training(ChatGLMForCausalLM 属性) training(CollieModelForCausalLM 属性) training(ColumnParallelLinearWithoutBias 属性) training(ColumnParallelLMHead 属性) training(GPTLMLoss 属性) training(InternLMForCausalLM 属性) training(LlamaForCausalLM 属性) training(Moss003MoonForCausalLM 属性) training(MossForCausalLM 属性) training(RowParallelLinearWithoutBias 属性) U update() (AccuracyMetric 方法) update() (BaseMetric 方法) update() (ClassifyFPreRecMetric 方法) update() (DecodeMetric 方法) update() (progress 方法) use_flash(CollieConfig 属性) W world_size(Env 属性) Z zero3_load_state_dict()(在 collie.utils 模块中)