lllyasviel / lllyasviel/ControlNet
train with multi-gpus
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 34.1k
- Forks
- 3k
- PR merge metrics
- No merged PRs in 30d
Description
@lllyasviel @williamyang1991 @eltociear @camenduru
hello bro,
Is it possible to train the tutorial_train_sd21.py with multi-GPUs?
I have modified the
`trainer = pl.Trainer(gpus=1, precision=32, callbacks=[logger])` to
`trainer = pl.Trainer(gpus=4, accelerator='dp,precision=32, callbacks=[logger])`
but it doesn't work
Can someone help me?
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with tutorial_train_sd21.py and the shown pl.Trainer configuration; reproduce the failure with the multi-GPU setting and inspect the reported error. Done when the tutorial training runs across four GPUs with a supported configuration.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 30/100