- 博客(4)
- 收藏
- 关注
RSS订阅原创 Pytorch加载模型
1、初始化 #使用GPU或者CPUdevice = torch.device('cuda' if cfg.CUDA else 'cpu’)#直接使用GPUdevice = torch.cuda.current_device() 2、加载模型#载入GPUpretrained_dict = torch.load(pretrained_path, map_location=lambda storage, loc: storage.cuda(device)) #载入CPUpretraine
2021-07-15 18:47:50
2557
1
原创 Annaconda安装opencv命令
python3.8conda install --channel https://conda.anaconda.org/menpo opencv
2021-04-07 10:32:19
345
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人