smalldiffusion 核心模块:diffusion.py
本文件是 smalldiffusion 的核心,包含噪声调度(Schedule)、训练循环(training_loop)和采样算法(samples),总计不到 100 行代码。
2.1 模块结构
PyTorch Dataset 体系技术文档
覆盖 map-style/IterableDataset、全部内置 Dataset 扩展、图数据与 HF datasets、典型项目扩展模式、padding 与 collate 职责划分,以及与 DataLoader 的衔接。
Pytorch 权重初始化方法
全面对比深度学习权重初始化方法的原理、公式推导、优缺点与适用场景,附 PyTorch 代码示例和 Transformer 架构初始化最佳实践。