Skip to content

关于v2v_edit任务的多卡问题 #6

Description

@fujiwara-ayako

作者您好!
我最近在使用您的模型进行v2v_edit任务,主要是把输入的视频中的某些主体去除。
但是由于我对视频的分辨率要求比较高,所以我设置的是1920*1080,81帧。
在这个情况下A100 80GB会OOM。由于我有四张这样的显卡,我在看过之前的一个关于分卡的issue后尝试按照您说的方法进行分卡,但是可能是我的方法有问题,分卡后数据似乎不能顺利流通,产出的结果是黑白马赛克视频。
因为我是一个刚刚入门这个领域的master program student,我怀疑是我对您的简短指示理解的不够透彻,可否烦请您详细一些地指教一下如何分卡?
我原本的分卡是在univideo_inference.py下进行的,具体来说是在transformer初始化之后,pipeline初始化之前把transformer的所有层平分给四个卡。
感谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions