- 博客(3)
- 资源 (1)
- 收藏
- 关注
转载 PyTorch中使用指定的GPU
PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。有如下两种方法来指定需要使用的GPU。1. 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定:CUDA_VISIBLE_DEVICES=1 python my_script.py1.2 python代码中设定:import osos.environ[...
2018-05-31 09:47:07 8076
转载 error: function "atomicAdd(double *, double)" has already been defined
在编译caffe-resnet-attention是遇到问题,在网上找到错误的原因,记录下来。 原文地址CUDA 8.0 provides a definition of atomicAdd on double quantities that was not present in previous CUDA toolkits. The code you are working with also ...
2018-05-30 11:08:11 491
转载 迁移学习与微调的关系
在实际中,因为数据量小的缘故,很少有人会去从零开始去训练出一个CNN(train from scratch)。相反,普遍的做法都是在一个大的数据集上(比如ImageNet,包含120万张来自1000的类的数据集)进行预训练一个CNN模型,然后在这使用这个模型作为固定特征提取器或者网络的初始化权重应用在特定的任务中。主要的迁移学习的场景:Ø CNN feature + classifier;这个比...
2018-05-29 10:55:59 921
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人