search

valeoai / Maskgit-pytorch

MIT License
145 stars 15 forks source link

How can I use my own dataset to train maskgit? #8

Closed zxb-0 closed 6 months ago

zxb-0 commented 6 months ago

@llvictorll How can I use my own dataset to train maskgit?

llvictorll commented 6 months ago

Hello, All you have to do is provide your own Pytorch dataset and update the method "get_data" here

Victor

zxb-0 commented 6 months ago

@llvictorll thank you very much