search

GuyTevet / motion-diffusion-model

The official PyTorch implementation of the paper "Human Motion Diffusion Model"
MIT License
3.06k stars 330 forks source link

About diffusion_steps 50 #201

Open buptxyb666 opened 4 months ago

buptxyb666 commented 4 months ago

Thanks for your great work. I wonder that how --diffusion_steps 50 influence the training process. Looking forward your reply !!

GuyTevet commented 4 months ago

It tunes the diffusion scheduler: https://github.com/GuyTevet/motion-diffusion-model/blob/dd0d0030b3659255fe9779e14852d2028269985f/utils/model_util.py#L55