search

takimailto / BerDiff

21 stars 1 forks source link

Hello,can you give us the training code? I don't know how to train this model. Should I use Probabilistic Unet's training code to train our model? #6

Closed sonnystalin closed 3 months ago

sonnystalin commented 4 months ago

Hello , thank you for your excellent work ,could you please give release the training code? I don't know how to train this model, and i'm looking forward to your early reply. Thank you! @takimailto

1234xcs commented 3 months ago

请问您运行 起来了吗 是用的啥运行的

takimailto commented 3 months ago

Thank you for reading the paper! The training code is similiar with the training code of improved diffusion (https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/train_util.py), and we will release our version recently.

1234xcs commented 3 months ago

很期待 你们的代码 希望近期能公布 祝你们好运

takimailto commented 3 months ago

Please refer to the newly added directory "training".

sonnystalin commented 3 months ago

Thank you for your excellent work!