yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
https://arxiv.org/abs/2011.13456
Apache License 2.0
1.58k stars 295 forks source link

Training VP-SDE #15

Open shreka116 opened 2 years ago

shreka116 commented 2 years ago

First of all, great thanks for such brilliant work!

I have one trivial question.

If I want to train a score network with VP-SDE on FFHQ datasets, what arguments do I need to change from "config/ve/ffhq_ncsnpp_continuous.py" ?

Best regards, Joseph