← 返回事件
持续讨论AI发版

trunk/38d3f6fd0d6e7ea5b68ee3854e1033ff08a5c92c: [FSDP2] Separate sharded and unsharded gradient dtypes (#194434)

发生了什么

Fixes #170648 feature 0 : when mp_policy.reduce_dtype is None, it follows orig model.parameters() .grad_dtype. .grad_dtype were silently ignored before this PR if orig grad_dtype is set, mp_policy.reduce_dtype is resolved to orig grad_dtype (including explicit None) if not set, mp_policy.reduce_dtype is resolved to orig param dtype this is BC breaking : mp_policy.reduce_dtype=None no longer defaults to mp_policy.par…

摘要按规则整理自下方来源原文

为什么在扩散

来源