Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
Published
20 hours ago •
aladdinpersson
aladdinpersson
DEVICE = "cuda" if torch.cuda.is_available else "cpu" not working on cpu machine
https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO/train.py
change that line to DEVICE = "cuda" if torch.cuda.is_available() else "cpu" works on cpu machine
