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

Reconstruct for colorization #22

Open itoshiyanazawa opened 1 year ago

itoshiyanazawa commented 1 year ago

I’m very interested in your work!

I’m considering to conduct colorization for my own dataset, but I’m not sure how to reconstruct your codes.

Does it work just change the code of your tutorial’s jupyter notebook?