训练篇

TorchEasyRec 训练阶段的完整流程,从初始化到模型构建、特征处理、嵌入管理、扩展开发、数据管线与采样器,以及训练附属组件(损失函数、评估指标、优化器)。

章节 标题 简介
04 初始化流程 从 CLI 到可训练模型的完整序列
05 训练流程 单步数据流、两阶段初始化、混合精度、Pipeline、optimizer 层级
06 训练部署 Docker 镜像、torchrun 启动、PAI-DLC 集成、分布式运行时、拓扑约束
07 模型系统 BaseModel 层级、20+ 模型目录、模型选择决策树、导出兼容矩阵
08 特征系统 12 种特征类型、4 种 FG 模式、纯 Python FG 迁移方案
09 嵌入系统 EmbeddingGroup、TorchRec 分片、ZCH、DenseEmbeddingCollection
10 DynamicEmb 集成 NVIDIA GPU 哈希表嵌入后端的双视角深度解析
11 损失函数 BinaryFocalLoss、JRCLoss、ParetoEfficientMultiTaskLoss
12 评估指标 GroupedAUC、XAUC、DecayAUC、Recall@K、NormalizedEntropy、UniqueRatio
13 优化器与学习率调度器 稀疏/稠密优化器类型、5 种 LR 调度器、混合精度与梯度配置
19 扩展机制与自定义模型 元类注册机制 vs if-elif 硬编码、完整扩展步骤、Proto 生成流程
20 数据管线:Dataset 体系 BaseDataset 基类、5 种 Dataset 类型、DataLoader 配置、Batch 数据结构
21 Sampler 采样体系 5 种采样器、graphlearn 框架、分布式协同、Batch 注入
22 Feature Group 与 Sequence Encoder 三种 group_type、5 种 Sequence Encoder、embedding_name_suffix

Table of contents


This site uses Just the Docs, a documentation theme for Jekyll.