How to solve the contradiction between Expected all tensors to be on the same device and can’t convert CUDA tensor to numpy

前端 未结 0 828
无人及你
无人及你 2021-01-04 13:57

I\'m trying yolov3 with multi GPUs... \'\'\'

def evaluate(self):
    self.models.eval()

    labels = []
    sample_metrics = []  # List of tuples (TP, confs,         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题