search

yuangan / EAT_code

Official code for ICCV 2023 paper: "Efficient Emotional Adaptation for Audio-Driven Talking-Head Generation".
Other
275 stars 31 forks source link

训练时间 #15

Closed YZX-codesky closed 10 months ago

YZX-codesky commented 10 months ago

您好，作者，非常感谢您的工作！我在训练第二阶段的代码的时候，发现训练的时间变得非常长，长达800多个小时，见下图： 训练截图 我采用的是一张3090显卡。想问一下这个是正常的吗？

yuangan commented 10 months ago

我们用了4张3090卡训练，按半个小时存一个epoch，一般来说第二阶段训练5-8个小时就行，具体训练时间和performance的关系可以参考我们论文中的Fig.5。1张3090可能需要训久一点，你可以根据训练loss和其他指标来判断需要训练多久。

YZX-codesky commented 10 months ago

我们用了4张3090卡训练，按半个小时存一个epoch，一般来说第二阶段训练5-8个小时就行，具体训练时间和performance的关系可以参考我们论文中的Fig.5。1张3090可能需要训久一点，你可以根据训练loss和其他指标来判断需要训练多久。

想问一下训练5-8个小时，是训练多少个epochs，是文件中的2000个吗？还是根据损失的收敛情况做了调整

image

YZX-codesky commented 10 months ago

想问一下训练5-8个小时，是训练多少个epochs，是文件中的2000个吗？还是根据损失的收敛情况做了调整

------------------ 原始邮件 ------------------ 发件人: "yuangan/EAT_code" @.>; 发送时间: 2024年1月8日(星期一) 下午2:29 @.>; @.**@.>; 主题: Re: [yuangan/EAT_code] 训练时间 (Issue #15)

我们用了4张3090卡训练，按半个小时存一个epoch，一般来说第二阶段训练5-8个小时就行，具体训练时间和performance的关系可以参考我们论文中的Fig.5。1张3090可能需要训久一点，你可以根据训练loss和其他指标来判断需要训练多久。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>