GitLab

utils.py中缺少get_device_name，导致调用出错。

def get_device_name(device:torch.device = None):
    if torch.cuda.is_available():
        if device is None:
            num_gpus = torch.cuda.device_count()
            gpu_name = []
            for gpu_id in range(num_gpus):
                gpu_name.append(torch.cuda.get_device_name(gpu_id))
            return gpu_name
        else:
            return torch.cuda.get_device_name(device)