请问下训练的配置

zhanglonghao1992 / One-Shot_Free-View_Neural_Talking_Head_Synthesis

Pytorch implementation of paper "One-Shot Free-View Neural Talking-Head Synthesis for Video Conferencing"

Other

764 stars 143 forks source link

Closed highway007 closed 2 years ago

highway007 commented 2 years ago

请问下大佬训练用的gpu是什么配置的啊？训练速度大概是多少？

zhanglonghao1992 commented 2 years ago

A-100 GPU x 8. 1 epoch = 1 hour.

highway007 commented 2 years ago

thanks！

vinayak015 commented 2 years ago

Could you tell approximately how many epochs were required for training? @zhanglonghao1992

zhanglonghao1992 commented 2 years ago

Could you tell approximately how many epochs were required for training? @zhanglonghao1992

About 120 epochs