-
-
Notifications
You must be signed in to change notification settings - Fork 744
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
how to train your inpainting model using my own dataset?? #352
Comments
from imagen_pytorch import Unet, Imagen, ImagenTrainer output_path="/content/drive/MyDrive/imgen_pytorch/output" unets for unconditional imagenunet = Unet( imagen, which contains the unet aboveimagen = Imagen( trainer = ImagenTrainer( instantiate your dataloader, which returns the necessary inputs to the DDPM as tuple in the order of images, text embeddings, then text masks. in this case, only images is returned as it is unconditional trainingdataset = Dataset('/content/drive/MyDrive/unconditional_generation/dataset_256', image_size = 256) trainer.add_train_dataset(dataset, batch_size = 16) working training loopfor i in range(20000):
This is the training code for your custom dataset . |
Thanks for sharing this amazing work,I want to train your inpainting model using my own dataset, could you show me any training script and how to prepare the data at your convenience?
The text was updated successfully, but these errors were encountered: