网站首页 model

• 

  pytorch中的model=model.to(device)使用说明

  这代表将模型加载到指定设备上。其中，device=torch.device("cpu")代表的使用cpu，而device=torch.device("cuda")则代表的使用GPU。当我们指定了设备之...

  2021-07-16 longtao100 python 1284 ℃ 0 评论