search

JQ-MC / object-detection-fedlearn

Repository of the conceptual work around Federated Learning to train the object detection model YOLOv5.
18 stars 4 forks source link

Dose this project support gpu? I have changed the device = 0 but it get errors. #4

Open gongzir1 opened 1 year ago

JQ-MC commented 1 year ago

The device in which you are running the project should support GPU if you set device = 0.

gongzir1 commented 1 year ago

I am running on gpu device but it got this error when I changed the device =0 : RuntimeError: don't know how to restore data location of torch.HalfStorage (tagged with 0)

gongzir1 commented 1 year ago

I changed it to "cuda:0" solved the problem

gongzir1 commented 1 year ago

Another thing is that I am confused about why it got 'best.pt' and evaluation results for each client's training instead of for each server.