New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Ignore the `freqs_cis` buffer so that DDP does not broadcast it at construction time #59

Open

xiaoguzai opened this issue Feb 15, 2024 · 0 comments

xiaoguzai commented Feb 15, 2024

在pretrain中有这样一部分代码

if ddp:
    # Ignore the `freqs_cis` buffer so that DDP does not broadcast it at
    # construction time since NCCL does not support `ComplexFloat`
    prefix = "_orig_mod." if compile else ""
    model._ddp_params_and_buffers_to_ignore = {prefix + "freqs_cis"}
    model = DDP(model, device_ids=[ddp_local_rank])

这里的操作作用是什么？有没有大佬指点一下？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment