Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Training procedure #15

Open
CharlesGong12 opened this issue Apr 24, 2024 · 1 comment
Open

Training procedure #15

CharlesGong12 opened this issue Apr 24, 2024 · 1 comment

Comments

@CharlesGong12
Copy link

Hi, thanks for your amazing work!
I find your work so interesting and I wanna train the models by myself. Could you provide training code?
@vidit98
Copy link
Collaborator

vidit98 commented May 10, 2024 •
edited
Loading

Hi, thanks for your interest! It would be difficult to release the training procedure soon. But I can help if you have any specific doubts. The training script would be very similar to LDM. The major change would be in dataloader where you need to read the segmentation mask as well.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@vidit98 @CharlesGong12