2018年03月_codebrid

12月 11月 10月 09月 08月 07月 06月 05月 04月 03月 02月 01月

原创 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