随笔分类 - pytorch
介绍pytorch的一些基础知识
摘要:解决RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda: https://blog.csdn.net/aminghhhh/article/details/1
阅读全文
摘要:参考:https://jiuaidu.com/jianzhan/574049/
阅读全文
摘要:固定随机种子的步骤请参考:https://zhengyujie.cn/2291.html 这里我主要讲一下模型在复现结果遇到的一些问题。 首先在固定随机种子之后,每次模型训练出来的结果一定是一样的,如果不相同,那么说明训练过程中一定还有一部分随机性操作没有被固定。 将模型参数保存下来,然后加载模型参
阅读全文
摘要:补充 现在安装显卡驱动和cuda,可以只需要执行两个命令即可,但是cuda的版本可能比较低,当然最准确的方式就是手动一步步安装。 在线安装驱动 sudo apt-get purge nvidia-* sudo apt-get update apt search nvidia-driver sudo
阅读全文
摘要:pytorch官方文档 :https://pytorch.org/docs/stable/index.html 1. pytorch的数学基本运算:https://blog.csdn.net/weicao1990/article/details/93738722 2. pytorch中一些向量的拼接
阅读全文
摘要:参考:https://blog.csdn.net/liujh845633242/article/details/102668515 这里我重点说一下1D卷积,2D卷积很好理解,但是1D卷积就不是那么好理解了,以textcnn为例,在对句子长度进行卷积之后,再将词向量的维度SUM成1维,总而言之,大家
阅读全文