DiffusionEngine
DiffusionEngine copied to clipboard
How to train my own DiffusionEngine?
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.
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.