- 博客(8)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 下载PyTorch
根据CUDA以及Python版本修改即可http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl
2019-04-18 20:24:54
3304
原创 PyTorch使用多GPU训练
1. PyTorch使用指定GPU训练model = Model()if torch.cuda.is_available(): model = model.cuda()上面代码直接使用model.cuda(), PyTorch默认使用从0开始的GPU, 有如下两种方法来指定需要使用的GPU:直接在终端中设定：CUDA_VISIBLE_DEVICES=1 python main....
2019-04-18 19:16:47
11444
1
原创 神经网络梯度消失与梯度爆炸
#神经网络梯度消失与梯度爆炸##为什么要使用梯度更新规则目前优化神经网络的方法都是基于反向传播的思想，即根据损失函数计算的误差通过梯度反向传播的方式，指导深度网络权值的更新优化。这样做是有一定原因的，首先，深层网络由许多非线性层堆叠而来，每一层非线性层都可以视为是一个非线性函数 (非线性来自于非线性激活函数），因此整个深度网络可以视为是一个复合的非线性多元函数...
2019-04-04 10:13:35
165
原创 PyTorch加载预训练模型
Pytorch调用预训练模型：1Pytorch保存和加载整个模型： torch.save(model, 'model.pth') model = torch.load('model.pth')Pytorch保存和加载预训练模型参数： torch.save(model.state_dict(), 'params.pth') model.load_state_dict(torc...
2018-11-15 20:27:54
15378
1
原创 行为识别
行为识别的数据库比较多，这里主要介绍两个最常用的数据库，也是近年这个方向的论文必做的数据库。 UCF101:来源为YouTube视频，共计101类动作，13320段视频。共有5个大类的动作：1)人-物交互；2)肢体运动；3)人-人交互；4)弹奏乐器；5)运动。数据库主页为：Center for Research in Computer Vision at the Univ...
2018-06-01 10:43:58
1509
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人