随笔分类 - 人工智能
摘要:cat /usr/local/cuda/version.txt cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
阅读全文
摘要:1、模型并行,in-graph replication;数据并行,between-graph replication。 tf.train.Supervisor tf.train.MonitoredTrainingSession 参考链接: https://github.com/tensorflow/
阅读全文
摘要:Baidu All Reduce,即Ring All Reduce。Ring All Reduce技术在高性能计算领域很常用,2017年被百度用于深度学习训练。 朴素All Reduce的通信时间随GPU节点数线性增长。Ring All Reduce的通信时间跟GPU节点数无关,只受限于GPU间最慢
阅读全文
摘要:torch.utils.data.Dataset是一个抽象类,表示一个数据集。自定义的数据集要继承这个类,并且重写__len__和__getitem__方法。 __len__:len(dataset)返回数据集的size。 __getitem__:支持索引,dataset[i]用来获取第i个样本。
阅读全文
摘要:RNN(Recurrent Neural Networks)循环神经网络 广泛应用于自然语言处理(NLP,Natural Language Processing) Back Propagation Through Time Vanishing gradient problem LSTM, 长期依赖,
阅读全文
摘要:参考链接 https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html
阅读全文
摘要:tensorflow模型可以利用tf.train.Saver类保存成文件。一个模型包含下面四个文件。 meta文件 存储计算图的protobuf。 data-00000-of-00001文件和index文件 存储权值和偏置的二进制文件。 checkpoint文件 存储模型checkpoint信息的文
阅读全文
摘要:9.0版本的变更 1)__global__函数不能重载。 2) 3) 4)CUDA环境变量增加CUDA_ENABLE_CRC_CHECK。 5)扭曲矩阵函数支持矩阵乘积:m=32, n=8, k=16;m=8, n=32, k=16;m=n=k=16。 6)增加新的统一内存节:系统分配器,硬件相干性
阅读全文
摘要:Mat cv::imread(const String & filename, int flags = IMREAD_COLOR) imwrite(filename, img[, params]) cvtColor(src, code[, dst[, dstCn]]) resize(src, dsi
阅读全文
摘要:numpy.concatenate((a1, a2, ...), axis=0, out=None) nan:not a number inf:infinate arange(start, stop, step):[start, stop),step是步长的数组 sin:正弦函数 cos:余弦函数
阅读全文
摘要:论文:https://arxiv.org/pdf/1711.07064.pdf 译文:使用条件对抗网络消除盲目运动导致的模糊 摘要 我们提出DeblurGAN,一种用于运动去模糊化的端到端的学习方法。这种学习基于条件对抗网络和内容损失。模型、代码和数据集都放在https://github.com/K
阅读全文
摘要:论文:https://arxiv.org/pdf/1512.03385.pdf 译文:用于图像识别的深度残差学习 摘要 越深的神经网络就越难训练。我们提出一种残差学习框架,来简化比之前深很多的网络的训练。 恒等映射(identity mapping)
阅读全文
摘要:论文:https://arxiv.org/pdf/1703.10593.pdf
阅读全文
摘要:论文:https://arxiv.org/pdf/1406.2661v1.pdf 译文:生成式对抗网络 摘要 我们提出一个通过对抗过程来评估生成模型的新框架。在这个框架里,我们同时训练两个模型:1)生成模型G,获取数据分布;2)判别模型D,评估一个样本是从训练集来的而不是模型G生成的概率。生成模型G
阅读全文
摘要:论文:https://arxiv.org/abs/1711.09020 译文:StarGAN:用于多域图像到图像翻译的统一生成对抗网络 摘要 最近有关图像到图像的两域翻译研究取得了很大的成功。但是,现有的方法在处理多域时,可扩展性和鲁棒性不行, 因为对于每对图像域,都要单独构建不同的模型。为了解决这
阅读全文
摘要:论文:https://arxiv.org/abs/1409.1556 译文:用于大规模图像识别的非常深的卷积网络 摘要 本文,我们研究,在大规模图像识别中,卷积网络的深度对准确度的影响。我们的主要贡献是,利用3*3的卷积核来增加网络的深度,并且做了全面的评估,当深度达到16~19个权值层时,已有最好
阅读全文
摘要:StyleGAN 参考链接 https://arxiv.org/pdf/1812.04948.pdf https://github.com/NVlabs/stylegan
阅读全文
摘要:SegNet,是一种基于编码器-解码器架构的深度全卷积神经网络,用于图像语义分割。 参考链接: https://ieeexplore.ieee.org/document/7803544
阅读全文

浙公网安备 33010602011771号