search

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.59k stars 294 forks source link

subVPSDE sample #8

Closed TLi347 closed 2 years ago

TLi347 commented 2 years ago

Hi,

When I use Score_SDE_demo_PyTorch.ipynb and set score-based model to subVPSDE, it shows that "AttributeError: 'subVPSDE' object has no attribute 'alphas'" in subsection "PC sampling", "PC inpainting", "PC colorizer".

TLi347 commented 2 years ago

sorry, it's mentioned in paper that only "For VE/VP SDEs, the drift coefficient is element-wise, and the diffusion coefficient is diagonal."