How to train my own DiffusionEngine?

[Lucy-IM]

Dear author, thanks for the great work. I want follow this work. if I want to train my own DiffusionEngine, how to set the dataset directory. What kind of data and tags do I need for training. Sorry for the interruption.

[mettyz]

Hi,

You can set the location for built-in datasets by export DETECTRON2_DATASETS=/path/to/datasets. If left unset, the default is ./datasets relative to your current working directory. The tutorial provided by [detectron2] may help.