How to train with single GPU? #27

Open

opened

作者你好，在accelerate_train.py中看到默认使用的是当前全部可用的GPU，怎么设置使其只用单个GPU呢？

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests