Skip to content

多卡并行时,queue的更新情况 #70

@DingHuilsp

Description

@DingHuilsp

代码中实现的,是不是多卡并行时,每个GPU的queue独立更新?,我感觉应该像moco一样,先keys = concat_all_gather(keys),再统一进行更新,是不是更合理一点?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions