Qiulin-W/SAFA

Checkpoints of the 3DMM pre-trained model

HowieMa opened this issue · 0 comments

Dear authors,
Thanks for sharing such a wonderful and interesting project! May I ask if you could release the pre-trained 3DMM model? i.e., the checkpoint after running the 3DMM Estimator Pre-training step with code

CUDA_VISIBLE_DEVICES="0,1,2,3" python -m torch.distributed.launch --nproc_per_node 4 run_ddp.py --config config/pretrain.yaml

As I'd like to run your end-to-end training step with a fully pre-trained 3DMM model given by "--tdmm_checkpoint".

Looking forward to your reply. Thanks in advance!