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