search

sail-sg / MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
Apache License 2.0
500 stars 35 forks source link

Train Custom Dataset #28

Closed xv994 closed 5 months ago

xv994 commented 5 months ago

Hey, I am coming too. infer_mdt.py works now, but when I trained my dataset, I had a new problem as below. image Do you guys know why it has a segmentation error? Please help me!

xv994 commented 5 months ago

When I ran the MDTv1, I met the same problem. image My GPU is 3090 24GB. batch_size = 1

gasvn commented 5 months ago

It seems like a environment issue. Can you try to reinstall your pytorch environment and also the cuda and cudnn?

liwenyang-911 commented 5 months ago

Can I add you to WeChat communication?

xv994 commented 5 months ago

Can I add you to WeChat communication?

give me your email address or wechat account

liwenyang-911 commented 5 months ago

Can I add you to WeChat communication?

give me your email address or wechat account

codetruth@163.com

xv994 commented 5 months ago

Can I add you to WeChat communication?

give me your email address or wechat account

codetruth@163.com

已发送

rutuja1409 commented 1 month ago

Hey, I want to train on my custom dataset. But its unconditional data set, just the images, no class id no annotations. Can I still train? How can I do that?

@gasvn @xv994 @liwenyang-911 @mavenlin @P2333 Can you please help me out here if you know?

xv994 commented 3 weeks ago

Hey, I want to train on my custom dataset. But its unconditional data set, just the images, no class id no annotations. Can I still train? How can I do that?

@gasvn @xv994 @liwenyang-911 @mavenlin @P2333 Can you please help me out here if you know?

I'm sorry for replying late.

This model is a supervised deep learning. Maybe using conditional dataset is not a right choice.