环境

  • GPU: Nvidia H800 * 1
  • CPU: 120 c
  • Memory: 80G

start command:

workspace=/gemini/code/Wav2Lip
cd $workspace
source /venv/bin/python
python wav2lip_train.py --data_root lrs2_preprocessed/ --checkpoint_dir wav2lip_checkpoints --syncnet_checkpoint_path syncnet_checkpoints/checkpoint_step000720000.pth

代码

git clone https://github.com/miiy/Wav2Lip.git
git checkout develop
pip install -r requirements.txt

处理数据集

训练 syncnet

训练 wav2lip