Training problems

[zhangshilong2000]

Hello, I have a question for you: does the model support training on homemade datasets? If so, what configuration files should be changed?

[Lirui-Zhao]

Hi,

Yes, the model supports training on homemade datasets. You will mainly need to modify the dataset_diffree.py file located here. This file handles the dataset processing, so you should adjust it according to the format of your homemade dataset.

If you have further questions, feel free to ask!

Best,
Lirui