search

Chris-hughes10 / Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.
GNU General Public License v3.0
116 stars 35 forks source link

Run on GPU #9

Closed nemeziz69 closed 1 year ago

nemeziz69 commented 1 year ago

How can I run the training with GPU?

Chris-hughes10 commented 1 year ago

Hi, this repo uses PyTorch-accelerated which handles training on the GPU. By default, just running the scripts using python ... will use 1 GPU if it is available.

For multiple GPUs, or setting mixed precision options, you can use the accelerate CLI commands to generate and use a config file, as described here.