search

labmlai / annotated_deep_learning_paper_implementations

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
https://nn.labml.ai
MIT License
54.48k stars 5.62k forks source link

Training code or references for training the latent diffusion model on a custom dataset #195

Open risejl opened 1 year ago

risejl commented 1 year ago

Hello! Thank you for your fantastic work!

I'm currently working on training the latent diffusion model using a custom dataset. I noticed that the repository doesn't include the training code for this specific model. Could you kindly share the training code or provide any references that would be suitable for training the latent diffusion model within the context of this repository?

Your guidance would be greatly appreciated!

TMDTom commented 11 months ago

Have find the suitable training code?We can have a communication. Thanks a lot!

HydrogenC commented 5 months ago

Also looking for training code :) I have no idea about how to train inpainting.

lyjlch commented 5 months ago

I wonder if there's any code that already works with the custom dataset, and if so, let me know and I'd appreciate it