DiffusionEngine icon indicating copy to clipboard operation
DiffusionEngine copied to clipboard

How to train my own DiffusionEngine?

Open Lucy-IM opened this issue 2 years ago • 1 comments

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.

Lucy-IM avatar Sep 14 '23 08:09 Lucy-IM

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.

mettyz avatar Sep 17 '23 03:09 mettyz