- 博客(11)
- 资源 (23)
- 收藏
- 关注
原创 cuda指定使用哪块GPU
$ nvidia-smi查看当前节点下有几块GPU,以及每一块的使用情况在运行程序时指定使用哪块GPU$ CUDA_VISIBLE_DEVICES=5 python main.py --cuda
2018-03-30 18:39:16 11342
原创 python zip()函数
python zip()函数 # print(len(data)) batch 个 tuple(torch.float(3,224,224), str) # print(data[0][0].size()) [3,224,224] # print(data[0][1].size()) [len] images, captions = z
2018-03-29 19:51:14 2192
转载 soft Attention 和Hard Attention
以下内容摘自:https://zhuanlan.zhihu.com/p/31547842 1.attention的提出:把输入X编码成一个固定的长度,对于句子中每个词都赋予相同的权重,这样是不合理的,没有区分度往往使模型性能下降。因此提出Attention Mechanism(注意力机制),用于对输入X的不同部分赋予不同的权重,进而实现软区分的目的。Kelvin Xu等人与2015年发...
2018-03-28 17:12:48 21344 2
原创 【error】invalid argument 1: must be strictly positive at /pytorch/torch/lib/TH/generic/THTenso
ERROR:RuntimeError: invalid argument 1: must be strictly positive at /pytorch/torch/lib/TH/generic/THTensorMath.c:2184具体: File "main.py", line 152, in test for batch_idx, (image, captions, len...
2018-03-25 10:19:47 3553 6
原创 【error】anaconda3/bin/activate: Permission denied
命令行输入$ activate py27报错如下-bash: /。。/anaconda3/bin/activate: Permission denied解决:$ source activate py27
2018-03-23 19:40:19 7736 1
原创 如何使用多块GPU
要把模型变成dataparallel的模型net1 = nn.DataParallel(net1)要注意的是,保存模型的时候要转回来torch.save(net2.cpu().module.state_dict(), )模型和Variable都要加.cuda()模型是model = torch.nn.DataParallel(model).cuda() 转换成
2018-03-21 19:32:53 2924
原创 MSCOCO数据集评估caption_evaluation
一、eval时,caption需准备格式如下(使用json.dump保存到文件):annotation{ "id": int, "image_id": int, "caption": str,}二、evaldemo/代码这里提供的评估代码可以用于获取公开可用的COCO验证集的结果。它计算多个常用指标,包括BLEU、METEOR、ROUGE-L和CIDEr(包含每个度量的引用和描述...
2018-03-21 14:20:37 9589 6
原创 为什么要用contiguous()
【error】input is not contiguous at....调用view之前最好先contiguousx.contiguous().view()因为view需要tensor的内存是整块的
2018-03-20 21:02:54 9402
原创 【Error】Error in `python': free(): invalid pointer: 0x00007f9b9de705e0
*** Error in `python': free(): invalid pointer: 0x00007f9b9de705e0 ***======= Backtrace: =========/lib64/libc.so.6(+0x7c619)[0x7f9c23a9c619]/users4/zsun/anaconda3/lib/python3.6/site-packages/torch/...
2018-03-19 17:47:33 6384
原创 image+caption+pytorch的简单demo(亲测好用)
在github上搜:https://github.com/search?utf8=%E2%9C%93&q=image+caption+pytorch&type=Repositories1. 源码:https://github.com/jinfagang/pytorch_image_caption目录:build_vocab建立vocab; resize将图像全部重新resize; ...
2018-03-19 14:26:40 4201 1
原创 【error】Resource 'tokenizers/punkt/PY3/english.pickle' not found. Please use the NLTK Downloader to
报错信息:LookupError: ********************************************************************** Resource 'tokenizers/punkt/PY3/english.pickle' not found. Please use the NLTK Downloader to obtain the r...
2018-03-13 10:19:10 4530
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人