search

THUDM / Inf-DiT

Official implementation of Inf-DiT: Upsampling Any-Resolution Image with Memory-Efficient Diffusion Transformer
Apache License 2.0
378 stars 19 forks source link

请问如何在自己创建的超分数据集上微调Inf-DiT? #29

Open XIAOXIXIXIXIXI opened 1 week ago

XIAOXIXIXIXIXI commented 1 week ago

感谢您出色的工作！ 现在我希望在我创建的自定义超分数据集上训练Inf-DiT，路径类似： ./my_dataset/hr/img_001.jpg...img_002.jpg.... ./my_dataset/lr/img_001.jpg...img_002.jpg.... 我注意到您目前使用的是WebDataset，我在将Inf-DiT应用到我自己的数据集上遇到了困难，希望得到您的指导