220 Star 945 Fork 694

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindflow.cell.DDPMPipeline.rst 875 Bytes
一键复制 编辑 原始数据 按行查看 历史
b_rookie 提交于 2025-04-02 15:46 +08:00 . [mod]: add attention and diffusion docs

mindflow.cell.DDPMPipeline

.. py:class:: mindflow.cell.DDPMPipeline(model, scheduler, batch_size, seq_len, num_inference_steps=1000, compute_dtype=mstype.float32)

    DDPM采样过程控制实现。

    参数:
        - **model** (nn.Cell) - 训练模型。
        - **scheduler** (DDPMScheduler) - 噪声控制器,用于去噪。
        - **batch_size** (int) - batch大小。
        - **seq_len** (int) - 序列长度。
        - **num_inference_steps** (int) - 采样的步数。默认值: ``1000`` 。
        - **compute_dtype** (mindspore.dtype) - 数据类型。默认值: ``mstype.float32`` ,表示 ``mindspore.float32`` 。

    异常:
        - **TypeError** - 如果 `scheduler` 不是 `DDPMScheduler` 类型。
        - **ValueError** - 如果 `num_inference_steps` 不等于 `scheduler.num_train_timesteps` 。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
r0.7

搜索帮助